EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Update 3dspeaker/export-onnx.py (#836)

Browse Source 
Update to match the changes in infer_sv.py at 3D-speaker. 

Added 2 more supported models and "zh_en" language.
This commit is contained in:
chiiyeh
2024-05-06 12:10:35 +08:00
committed by GitHub
parent 4f758e6cd3
commit 9c8255fdb2
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/3dspeaker/export-onnx.py
View File

@@ -42,11 +42,13 @@ def get_args():
choices=[ choices=[
"speech_campplus_sv_en_voxceleb_16k", "speech_campplus_sv_en_voxceleb_16k",
"speech_campplus_sv_zh-cn_16k-common", "speech_campplus_sv_zh-cn_16k-common",
"speech_campplus_sv_zh_en_16k-common_advanced",
"speech_eres2net_sv_en_voxceleb_16k", "speech_eres2net_sv_en_voxceleb_16k",
"speech_eres2net_sv_zh-cn_16k-common", "speech_eres2net_sv_zh-cn_16k-common",
"speech_eres2net_base_200k_sv_zh-cn_16k-common", "speech_eres2net_base_200k_sv_zh-cn_16k-common",
"speech_eres2net_base_sv_zh-cn_3dspeaker_16k", "speech_eres2net_base_sv_zh-cn_3dspeaker_16k",
"speech_eres2net_large_sv_zh-cn_3dspeaker_16k", "speech_eres2net_large_sv_zh-cn_3dspeaker_16k",
"speech_eres2netv2_sv_zh-cn_16k-common",
], ],
) )
return parser.parse_args() return parser.parse_args()
@@ -56,7 +58,7 @@ def get_args():
def main(): def main():
args = get_args() args = get_args()
local_model_dir = "pretrained" local_model_dir = "pretrained"
model_id = f"damo/{args.model}" model_id = f"iic/{args.model}"
conf = supports[model_id] conf = supports[model_id]
cache_dir = snapshot_download( cache_dir = snapshot_download(
model_id, model_id,
@@ -121,13 +123,15 @@ def main():
if "zh-cn" in args.model: if "zh-cn" in args.model:
language = "Chinese" language = "Chinese"
elif "zh_en" in args.model:
language = "Chinese-English"
elif "en" in args.model: elif "en" in args.model:
language = "English" language = "English"
else: else:
raise ValueError(f"Unsupported language for model {args.model}") raise ValueError(f"Unsupported language for model {args.model}")
comment = f"This model is from damo/{args.model}" comment = f"This model is from iic/{args.model}"
url = f"https://www.modelscope.cn/models/damo/{args.model}/summary" url = f"https://www.modelscope.cn/models/iic/{args.model}/summary"
meta_data = { meta_data = {
"framework": "3d-speaker", "framework": "3d-speaker",