跳转至

关键词检测模型

目前我们提供了中文和英文两个基础模型,都支持 Pytorch 和 onnxruntime 框架,Pytorch 模型主要用于微调,onnx 模型主要用于部署,可先使用 onnx 测试目标关键词的效果,如果达不到预期再考虑基于我们提供的基础模型微调。

语言 推理框架 下载地址 使用方法 简介
中文 Pytorch github 训练及微调方法 该模型是基于 Wenetspeech 1万小时训练,模型参数约为 3.3M,拼音(声韵母)建模,可作为基础模型用于微调。
中文 onnxruntime github modelscope 部署教程 此为上述基础模型导出的 onnx,可用于 sherpa-onnx 平台部署
英文 Pytorch github 训练及微调方法 该模型是基于 Gigaspeech 1万小时训练,模型参数约为 3.3M,BPE 建模,可作为基础模型用于微调。
英文 onnxruntime github modelscope 部署教程 此为上述基础模型导出的 onnx,可用于 sherpa-onnx 平台部署

如果上述下载链接(尤其是 github/huggingface 链接)被墙无法下载,可使用我们的提供的下载工具下载。

评论

如果您通过 github 登录评论有困难,可以在留言板把你的问题告诉我们。
您也可以加入我们的微信QQ群与广大开发者一起交流,也欢迎大家关注我们的微信公众号