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

Update asr_utils.py

Browse Source
This commit is contained in:
hnluo 2023-02-20 15:01:50 +08:00 committed by GitHub
parent 7064e8f36e
commit 4a796298cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
funasr/utils/asr_utils.py
View File

@ -27,7 +27,7 @@ def get_version():
def sample_rate_checking(audio_in: Union[str, bytes], audio_format: str):
r_audio_fs = None
if audio_format == 'wav':
if audio_format == 'wav' or audio_format == 'scp':
r_audio_fs = get_sr_from_wav(audio_in)
elif audio_format == 'pcm' and isinstance(audio_in, bytes):
r_audio_fs = get_sr_from_bytes(audio_in)
@ -134,6 +134,13 @@ def get_sr_from_wav(fname: str):
fs = None
else:
audio, fs = torchaudio.load(fname)
elif audio_type == "scp":
with open(fname, encoding="utf-8") as f:
for line in f:
wav_path = line.split()[1]
fs = get_sr_from_wav(wav_path)
if fs is not None:
break
return fs
elif os.path.isdir(fname):
dir_files = os.listdir(fname)