1
0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-05 01:10:22 +08:00

optimize(rvc.f0): rename inner defs

This commit is contained in:
源文雨
2024-06-13 00:51:22 +09:00
parent 83144868e1
commit 1e94e007d5
6 changed files with 6 additions and 7 deletions

View File

@@ -10,7 +10,6 @@ from time import time
import faiss
import librosa
import numpy as np
import pyworld
import torch
import torch.nn.functional as F
import torchcrepe

View File

@@ -28,4 +28,4 @@ class Dio(F0Predictor):
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate)
for index, pitch in enumerate(f0):
f0[index] = round(pitch, 1)
return self.interpolate_f0(self.resize_f0(f0, p_len))[0]
return self._interpolate_f0(self._resize_f0(f0, p_len))[0]

View File

@@ -17,7 +17,7 @@ class F0Predictor(object):
filter_radius: Optional[Union[int, float]] = None,
): ...
def interpolate_f0(self, f0: np.ndarray[Any, np.dtype]):
def _interpolate_f0(self, f0: np.ndarray[Any, np.dtype]):
"""
对F0进行插值处理
"""
@@ -55,7 +55,7 @@ class F0Predictor(object):
return ip_data[:, 0], vuv_vector[:, 0]
def resize_f0(self, x: np.ndarray[Any, np.dtype], target_len: int):
def _resize_f0(self, x: np.ndarray[Any, np.dtype], target_len: int):
source = np.array(x)
source[source < 0.001] = np.nan
target = np.interp(

View File

@@ -29,4 +29,4 @@ class Harvest(F0Predictor):
f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate)
if filter_radius is not None and filter_radius > 2:
f0 = signal.medfilt(f0, 3)
return self.interpolate_f0(self.resize_f0(f0, p_len))[0]
return self._interpolate_f0(self._resize_f0(f0, p_len))[0]

View File

@@ -36,4 +36,4 @@ class PM(F0Predictor):
pad_size = (p_len - len(f0) + 1) // 2
if pad_size > 0 or p_len - len(f0) - pad_size > 0:
f0 = np.pad(f0, [[pad_size, p_len - len(f0) - pad_size]], mode="constant")
return self.interpolate_f0(f0)[0]
return self._interpolate_f0(f0)[0]

View File

@@ -122,7 +122,7 @@ class RMVPE(F0Predictor):
f0 = self._decode(hidden, thred=filter_radius)
return self.interpolate_f0(self.resize_f0(f0, p_len))[0]
return self._interpolate_f0(self._resize_f0(f0, p_len))[0]
def _to_local_average_cents(self, salience, threshold=0.05):
center = np.argmax(salience, axis=1) # 帧长#index