Jelajahi Sumber

unit test modelscope for sd

志浩 3 tahun lalu
induk
melakukan
92ca6a31cf
1 mengubah file dengan 24 tambahan dan 0 penghapusan
  1. 24 0
      egs/alimeeting/diarization/sond/unit_test_modelscope.py

+ 24 - 0
egs/alimeeting/diarization/sond/unit_test_modelscope.py

@@ -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={},