Explorar el Código

Update dataset.py

fix dest_sample_rate bug
zhuzizyf hace 3 años
padre
commit
2cdb2d654f
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      funasr/datasets/dataset.py

+ 1 - 1
funasr/datasets/dataset.py

@@ -115,7 +115,7 @@ def sound_loader(path, dest_sample_rate=16000, float_dtype=None):
     # NOTE(kamo): SoundScpReader doesn't support pipe-fashion
     # NOTE(kamo): SoundScpReader doesn't support pipe-fashion
     # like Kaldi e.g. "cat a.wav |".
     # like Kaldi e.g. "cat a.wav |".
     # NOTE(kamo): The audio signal is normalized to [-1,1] range.
     # NOTE(kamo): The audio signal is normalized to [-1,1] range.
-    loader = SoundScpReader(path, dest_sample_rate=16000, normalize=True, always_2d=False)
+    loader = SoundScpReader(path, dest_sample_rate, normalize=True, always_2d=False)
 
 
     # SoundScpReader.__getitem__() returns Tuple[int, ndarray],
     # SoundScpReader.__getitem__() returns Tuple[int, ndarray],
     # but ndarray is desired, so Adapter class is inserted here
     # but ndarray is desired, so Adapter class is inserted here