From dae8f7472dee1f630d45828c186f028c05cb298f Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 25 二月 2023 19:42:25 +0800
Subject: [PATCH] Merge pull request #153 from alibaba-damo-academy/dev_gzf
---
funasr/utils/asr_utils.py | 9 ++++++++-
1 files changed, 8 insertions(+), 1 deletions(-)
diff --git a/funasr/utils/asr_utils.py b/funasr/utils/asr_utils.py
index a3ff3e3..0f0e4c3 100644
--- a/funasr/utils/asr_utils.py
+++ b/funasr/utils/asr_utils.py
@@ -27,7 +27,7 @@
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 @@
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)
--
Gitblit v1.9.1