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.
正在显示
1 个修改的文件
包含
7 行增加
和
3 行删除
| @@ -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", |
-
请 注册 或 登录 后发表评论