Mirrors/Mangio-RVC-Fork
1
0
Fork 0
mirror of https://github.com/Mangio621/Mangio-RVC-Fork.git synced 2025-12-16 03:27:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

add rmvpe to hybrid implementation

Browse Source
This commit is contained in:
Sarah
2023-08-04 08:41:47 +02:00
parent 7c46f9ccde
commit 112bc366aa
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vc_infer_pipeline.py
View File

@@ -224,6 +224,16 @@ class VC(object):
if filter_radius > 2: if filter_radius > 2:
f0 = signal.medfilt(f0, 3) f0 = signal.medfilt(f0, 3)
f0 = f0[1:] # Get rid of first frame. f0 = f0[1:] # Get rid of first frame.
elif method == "rmvpe":
if hasattr(self, "model_rmvpe") == False:
from rmvpe import RMVPE
print("loading rmvpe model")
self.model_rmvpe = RMVPE(
"rmvpe.pt", is_half=self.is_half, device=self.device
)
f0 = self.model_rmvpe.infer_from_audio(x, thred=0.03)
f0 = f0[1:] # Get rid of first frame.
elif method == "dio": # Potentially buggy? elif method == "dio": # Potentially buggy?
f0, t = pyworld.dio( f0, t = pyworld.dio(
x.astype(np.double), x.astype(np.double),