mirror of
https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git
synced 2026-06-06 01:30:24 +08:00
optimize(rmvpe): move rmvpe into rvc.f0
This commit is contained in:
3
tools/cmd/onnx/export.py
Normal file
3
tools/cmd/onnx/export.py
Normal file
@@ -0,0 +1,3 @@
|
||||
from rvc.onnx import export_onnx
|
||||
|
||||
export_onnx("pt/Justin Bieber.pth", "pt/TestRvc_Rvc.onnx")
|
||||
22
tools/cmd/onnx/infer.py
Normal file
22
tools/cmd/onnx/infer.py
Normal file
@@ -0,0 +1,22 @@
|
||||
import soundfile
|
||||
import librosa
|
||||
|
||||
from rvc.onnx import RVC
|
||||
|
||||
hop_size = 512
|
||||
sampling_rate = 40000 # 采样率
|
||||
f0_up_key = 0 # 升降调
|
||||
sid = 0 # 角色ID
|
||||
f0_method = "dio" # F0提取算法
|
||||
model_path = "exported_model.onnx" # 模型的完整路径
|
||||
vec_path = "vec-256-layer-9.onnx" # 需要onnx的vec模型
|
||||
wav_path = "123.wav" # 输入路径或ByteIO实例
|
||||
out_path = "out.wav" # 输出路径或ByteIO实例
|
||||
|
||||
model = RVC(model_path, vec_path=vec_path, hop_len=hop_size, device="cuda")
|
||||
|
||||
wav, sr = librosa.load(wav_path, sr=sampling_rate)
|
||||
|
||||
audio = model.infer(wav, sr, sampling_rate, sid, f0_method, f0_up_key)
|
||||
|
||||
soundfile.write(out_path, audio, sampling_rate)
|
||||
Reference in New Issue
Block a user