import numpy as np import pyworld from modules.F0Predictor.F0Predictor import F0Predictor class HarvestF0Predictor(F0Predictor): def __init__(self,hop_length=512,f0_min=50,f0_max=1100,sampling_rate=44100): self.hop_length = hop_length self.f0_min = f0_min self.f0_max = f0_max self.sampling_rate = sampling_rate self.name = "harvest" def interpolate_f0(self,f0): ''' 对F0进行插值处理 ''' vuv_vector = np.zeros_like(f0, dtype=np.float32) vuv_vector[f0 > 0.0] = 1.0 vuv_vector[f0 <= 0.0] = 0.0 nzindex = np.nonzero(f0)[0] data = f0[nzindex] nzindex = nzindex.astype(np.float32) time_org = self.hop_length / self.sampling_rate * nzindex time_frame = np.arange(f0.shape[0]) * self.hop_length / self.sampling_rate if data.shape[0] <= 0: return np.zeros(f0.shape[0], dtype=np.float32),vuv_vector if data.shape[0] == 1: return np.ones(f0.shape[0], dtype=np.float32) * f0[0],vuv_vector f0 = np.interp(time_frame, time_org, data, left=data[0], right=data[-1]) return f0,vuv_vector def resize_f0(self,x, target_len): source = np.array(x) source[source<0.001] = np.nan target = np.interp(np.arange(0, len(source)*target_len, len(source))/ target_len, np.arange(0, len(source)), source) res = np.nan_to_num(target) return res def compute_f0(self,wav,p_len=None): if p_len is None: p_len = wav.shape[0]//self.hop_length f0, t = pyworld.harvest( wav.astype(np.double), fs=self.hop_length, f0_ceil=self.f0_max, f0_floor=self.f0_min, frame_period=1000 * self.hop_length / self.sampling_rate, ) f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.fs) return self.interpolate_f0(self.resize_f0(f0, p_len))[0] def compute_f0_uv(self,wav,p_len=None): if p_len is None: p_len = wav.shape[0]//self.hop_length f0, t = pyworld.harvest( wav.astype(np.double), fs=self.sampling_rate, f0_floor=self.f0_min, f0_ceil=self.f0_max, frame_period=1000 * self.hop_length / self.sampling_rate, ) f0 = pyworld.stonemask(wav.astype(np.double), f0, t, self.sampling_rate) return self.interpolate_f0(self.resize_f0(f0, p_len))