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

unit test modelscope for sd

Browse Source
This commit is contained in:
志浩 2023-02-27 16:18:38 +08:00
parent bc145655e4
commit 92ca6a31cf
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
egs/alimeeting/diarization/sond/unit_test_modelscope.py
View File

@ -52,6 +52,30 @@ def test_without_profile_gpu_infer():
task=Tasks.speaker_diarization,
model='damo/speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch',
mode="sond_demo",
sv_model="damo/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch",
sv_model_revision="master",
num_workers=0,
log_level="WARNING",
param_dict={},
)
results = diar_pipeline(raw_inputs=raw_inputs)
print(results)
def test_url_without_profile_gpu_infer():
raw_inputs = [
"https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/speaker_diarization/record.wav",
"https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/speaker_diarization/spk1.wav",
"https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/speaker_diarization/spk2.wav",
"https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/speaker_diarization/spk3.wav",
"https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_data/speaker_diarization/spk4.wav",
]
diar_pipeline = pipeline(
task=Tasks.speaker_diarization,
model='damo/speech_diarization_sond-zh-cn-alimeeting-16k-n16k4-pytorch',
mode="sond_demo",
sv_model="damo/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch",
sv_model_revision="master",
num_workers=0,
log_level="WARNING",
param_dict={},