仁迷 3 лет назад
Родитель
Сommit
ff78a5ea80
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      funasr/datasets/large_datasets/dataset.py

+ 2 - 1
funasr/datasets/large_datasets/dataset.py

@@ -1,6 +1,6 @@
 import os
 import random
-import soundfile
+import numpy
 from functools import partial
 
 import torch
@@ -119,6 +119,7 @@ class AudioDataset(IterableDataset):
                     elif data_type == "sound":
                         key, path = item.strip().split()
                         waveform, sampling_rate = torchaudio.load(path)
+                        waveform = waveform.numpy()
                         mat = waveform[0]
                         sample_dict[data_name] = mat
                         sample_dict["sampling_rate"] = sampling_rate