FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
shixian.shi
2023-11-23 adc88bd9e76644badbbe006913addfa7cbe5d89c 
 funasr/datasets/iterable_dataset.py


@@ -14,7 +14,8 @@


import numpy as np


import torch


import torchaudio


import soundfile


# import librosa


import librosa


from torch.utils.data.dataset import IterableDataset


import os.path




@@ -70,7 +71,8 @@


    try:


        return torchaudio.load(input)[0].numpy()


    except:


        waveform, _ = soundfile.read(input, dtype='float32')


        # waveform, _ = librosa.load(input, dtype='float32')


        waveform, _ = librosa.load(input, dtype='float32')


        if waveform.ndim == 2:


            waveform = waveform[:, 0]


        return np.expand_dims(waveform, axis=0)