speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

wav is not exist

Browse Source
This commit is contained in:
游雁 2024-05-02 13:03:24 +08:00
parent f690741b9a
commit 0f7ce31c2a
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
funasr/datasets/sense_voice_datasets/datasets.py
View File

@ -2,7 +2,7 @@ import logging
import torch
import random
import traceback
from funasr.register import tables
from funasr.utils.load_utils import extract_fbank, load_audio_text_image_video
@ -73,15 +73,17 @@ class SenseVoiceDataset(torch.utils.data.Dataset):
if idx == 0:
index_cur = index
else:
if index <= self.retry:
index_cur = index + idx
else:
index_cur = torch.randint(0, index, ()).item()
index_cur = torch.randint(0, len(self.index_ds), ()).item()
item = self.index_ds[index_cur]
source = item["source"]
data_src = load_audio_text_image_video(source, fs=self.fs)
try:
data_src = load_audio_text_image_video(source, fs=self.fs)
except Exception as e:
logging.error(f"Loading wav failed! {str(e)}, {traceback.format_exc()}")
continue
if self.preprocessor_speech:
data_src = self.preprocessor_speech(data_src, fs=self.fs)
speech, speech_lengths = extract_fbank(