From 31dda90f2dd9fcb7410422228888dc7f8ff3c012 Mon Sep 17 00:00:00 2001
From: 志浩 <neo.dzh@alibaba-inc.com>
Date: 星期五, 03 三月 2023 10:45:10 +0800
Subject: [PATCH] modify infer.py
---
egs_modelscope/speaker_verification/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/infer.py | 16 +++++++++++-----
1 files changed, 11 insertions(+), 5 deletions(-)
diff --git a/egs_modelscope/speaker_verification/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/infer.py b/egs_modelscope/speaker_verification/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/infer.py
index e81297a..a48088c 100644
--- a/egs_modelscope/speaker_verification/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/infer.py
+++ b/egs_modelscope/speaker_verification/speech_xvector_sv-zh-cn-cnceleb-16k-spk3465-pytorch/infer.py
@@ -9,17 +9,23 @@
)
# 鎻愬彇涓嶅悓鍙ュ瓙鐨勮璇濅汉宓屽叆鐮�
+ # for url use "utt_id" as key
rec_result = inference_sv_pipline(
audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/sv_example_enroll.wav')
- enroll = rec_result["spk_embedding"]
+ enroll = rec_result["utt_id"]
- rec_result = inference_sv_pipline(
- audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/sv_example_same.wav')
- same = rec_result["spk_embedding"]
+ # for local file use "utt_id" as key
+ rec_result = inference_sv_pipline(audio_in='sv_example_same.wav')["test1"]
+ same = rec_result["test1"]
+
+ import soundfile
+ wav = soundfile.read('sv_example_enroll.wav')[0]
+ # for raw inputs use "utt_id" as key
+ spk_embedding = inference_sv_pipline(audio_in=wav)["utt_id"]
rec_result = inference_sv_pipline(
audio_in='https://isv-data.oss-cn-hangzhou.aliyuncs.com/ics/MaaS/ASR/test_audio/sv_example_different.wav')
- different = rec_result["spk_embedding"]
+ different = rec_result["utt_id"]
# 瀵圭浉鍚岀殑璇磋瘽浜鸿绠椾綑寮︾浉浼煎害
sv_threshold = 0.9465
--
Gitblit v1.9.1