File size: 2,949 Bytes
b15e31b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | # -*- coding: utf-8 -*-
"""Subprocess runner for vendored upstream RVC conversion."""
from __future__ import annotations
import argparse
import logging
import os
import re
import sys
from pathlib import Path
import soundfile as sf
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Run vendored upstream RVC VC")
parser.add_argument("--sid", required=True)
parser.add_argument("--vocals-path", required=True)
parser.add_argument("--output-path", required=True)
parser.add_argument("--f0-method", required=True)
parser.add_argument("--pitch-shift", type=int, required=True)
parser.add_argument("--index-path", default="")
parser.add_argument("--index-rate", type=float, required=True)
parser.add_argument("--filter-radius", type=int, required=True)
parser.add_argument("--rms-mix-rate", type=float, required=True)
parser.add_argument("--protect", type=float, required=True)
parser.add_argument("--speaker-id", type=int, required=True)
return parser.parse_args()
def main() -> int:
args = parse_args()
repo_root = Path(__file__).resolve().parent.parent
official_root = repo_root / "_official_rvc"
logging.basicConfig(level=logging.INFO, format="%(asctime)s | %(levelname)s | %(name)s | %(message)s")
sys.path.insert(0, str(official_root))
os.chdir(official_root)
sys.argv = [sys.argv[0]]
from configs.config import Config # type: ignore
from infer.modules.vc.modules import VC # type: ignore
config = Config()
vc = VC(config)
vc.get_vc(args.sid)
spk_max = 1
try:
if getattr(vc, "cpt", None) is not None:
spk_max = int(vc.cpt["config"][-3])
except Exception:
spk_max = 1
spk_id = max(0, min(max(1, spk_max) - 1, int(args.speaker_id)))
info, (sr, audio) = vc.vc_single(
spk_id,
args.vocals_path,
args.pitch_shift,
None,
args.f0_method,
args.index_path,
"",
args.index_rate,
args.filter_radius,
0,
args.rms_mix_rate,
args.protect,
)
if sr is None or audio is None:
raise RuntimeError(info)
output_path = Path(args.output_path)
output_path.parent.mkdir(parents=True, exist_ok=True)
sf.write(output_path, audio, sr)
match = re.search(
r"Time:\s*npy:\s*([0-9.]+)s,\s*f0:\s*([0-9.]+)s,\s*infer:\s*([0-9.]+)s\.",
str(info),
re.IGNORECASE | re.MULTILINE,
)
print("转换成功。", flush=True)
if args.index_path:
print(f"索引：{args.index_path}", flush=True)
if match:
print(
f"耗时：npy {match.group(1)}s，f0 {match.group(2)}s，推理 {match.group(3)}s",
flush=True,
)
return 0
if __name__ == "__main__":
raise SystemExit(main())
|