From 4a796298cca53fcc2c132e75b5381951a15e1a31 Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期一, 20 二月 2023 15:01:50 +0800
Subject: [PATCH] Update asr_utils.py
---
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