Unverified Commit f62a3d9d authored by SWHL's avatar SWHL Committed by GitHub
Browse files

Fix bugs

parent 7d0435cd
...@@ -270,7 +270,7 @@ def compute_fbank_feats( ...@@ -270,7 +270,7 @@ def compute_fbank_feats(
waveform, waveform,
blackman_coeff=0.42, blackman_coeff=0.42,
dither=1.0, dither=1.0,
energy_floor=0.0, energy_floor=1.0,
frame_length=25, frame_length=25,
frame_shift=10, frame_shift=10,
high_freq=0, high_freq=0,
......
...@@ -55,7 +55,7 @@ class RapidParaformer(): ...@@ -55,7 +55,7 @@ class RapidParaformer():
def load_data(self, def load_data(self,
wav_content: Union[str, np.ndarray, List[str]]) -> List: wav_content: Union[str, np.ndarray, List[str]]) -> List:
def load_wav(path: str) -> np.ndarray: def load_wav(path: str) -> np.ndarray:
waveform, _ = librosa.load(path) waveform, _ = librosa.load(path, sr=None)
return waveform[None, ...] return waveform[None, ...]
if isinstance(wav_content, np.ndarray): if isinstance(wav_content, np.ndarray):
......
...@@ -165,6 +165,7 @@ class WavFrontend(): ...@@ -165,6 +165,7 @@ class WavFrontend():
frame_shift=self.frame_shift, frame_shift=self.frame_shift,
dither=self.dither, dither=self.dither,
energy_floor=0.0, energy_floor=0.0,
window_type=self.window_type,
sample_frequency=self.fs) sample_frequency=self.fs)
feat = mat.astype(np.float32) feat = mat.astype(np.float32)
feat_len = np.array(mat.shape[0]).astype(np.int32) feat_len = np.array(mat.shape[0]).astype(np.int32)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment