wenet online onnx decoder
仓库地址
与文章魔改WeNet导出Onnx关联
准备onnx模型
本系统使用wenet-onnx导出的onnx模型
- encoder_chunk.onnx
- decoder.onnx
- ctc.onnnx
环境配置
torch、CUDA版本以及其他python包安装,参考wenet官方文档:https://github.com/mobvoi/wenet
文件配置
根据model_onnx_template.yaml中的描述补全lang_char.txt关键文件和onnx_model文件夹,默认模型为16k采样率数据训练所得。
系统运行
在正确的环境下直接执行python wenet_online_decoder_onnx.py,执行offline的识别 和 读取wav文件仿流式识别。
参考
本代码参考wenet_onlinedecode进行修改