chiiyeh
Committed by GitHub

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

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

Added 2 more supported models and "zh_en" language.
@@ -42,11 +42,13 @@ def get_args(): @@ -42,11 +42,13 @@ def get_args():
42 choices=[ 42 choices=[
43 "speech_campplus_sv_en_voxceleb_16k", 43 "speech_campplus_sv_en_voxceleb_16k",
44 "speech_campplus_sv_zh-cn_16k-common", 44 "speech_campplus_sv_zh-cn_16k-common",
  45 + "speech_campplus_sv_zh_en_16k-common_advanced",
45 "speech_eres2net_sv_en_voxceleb_16k", 46 "speech_eres2net_sv_en_voxceleb_16k",
46 "speech_eres2net_sv_zh-cn_16k-common", 47 "speech_eres2net_sv_zh-cn_16k-common",
47 "speech_eres2net_base_200k_sv_zh-cn_16k-common", 48 "speech_eres2net_base_200k_sv_zh-cn_16k-common",
48 "speech_eres2net_base_sv_zh-cn_3dspeaker_16k", 49 "speech_eres2net_base_sv_zh-cn_3dspeaker_16k",
49 "speech_eres2net_large_sv_zh-cn_3dspeaker_16k", 50 "speech_eres2net_large_sv_zh-cn_3dspeaker_16k",
  51 + "speech_eres2netv2_sv_zh-cn_16k-common",
50 ], 52 ],
51 ) 53 )
52 return parser.parse_args() 54 return parser.parse_args()
@@ -56,7 +58,7 @@ def get_args(): @@ -56,7 +58,7 @@ def get_args():
56 def main(): 58 def main():
57 args = get_args() 59 args = get_args()
58 local_model_dir = "pretrained" 60 local_model_dir = "pretrained"
59 - model_id = f"damo/{args.model}" 61 + model_id = f"iic/{args.model}"
60 conf = supports[model_id] 62 conf = supports[model_id]
61 cache_dir = snapshot_download( 63 cache_dir = snapshot_download(
62 model_id, 64 model_id,
@@ -121,13 +123,15 @@ def main(): @@ -121,13 +123,15 @@ def main():
121 123
122 if "zh-cn" in args.model: 124 if "zh-cn" in args.model:
123 language = "Chinese" 125 language = "Chinese"
  126 + elif "zh_en" in args.model:
  127 + language = "Chinese-English"
124 elif "en" in args.model: 128 elif "en" in args.model:
125 language = "English" 129 language = "English"
126 else: 130 else:
127 raise ValueError(f"Unsupported language for model {args.model}") 131 raise ValueError(f"Unsupported language for model {args.model}")
128 132
129 - comment = f"This model is from damo/{args.model}"  
130 - url = f"https://www.modelscope.cn/models/damo/{args.model}/summary" 133 + comment = f"This model is from iic/{args.model}"
  134 + url = f"https://www.modelscope.cn/models/iic/{args.model}/summary"
131 135
132 meta_data = { 136 meta_data = {
133 "framework": "3d-speaker", 137 "framework": "3d-speaker",