Commit a6b312ec authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 8f79d71e
# asr_onnx # asr
## 模型介绍
语音识别模型asr,speech-to-text(stt) 语音识别模型asr,speech-to-text(stt)
## 模型结构 ## 模型结构
![image](https://developer.hpccube.com/codes/modelzoo/asr_onnx/-/raw/main/resources/silero_stt_model.jpg) ![image](https://developer.hpccube.com/codes/modelzoo/asr_onnx/-/raw/main/resources/silero_stt_model.jpg)
## 推理 ## 算法原理
### 环境配置 ### 环境配置
[光源](https://www.sourcefind.cn/#/service-details)可拉取推理的docker镜像,在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载onnxruntime安装包。asr_onnx推荐的镜像如下: [光源](https://www.sourcefind.cn/#/service-details)可拉取推理的docker镜像,在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载onnxruntime安装包。asr_onnx推荐的镜像如下:
``` ```
...@@ -16,12 +16,14 @@ wget http://10.0.50.210:8000/jenkins/rocm/23.04.1/centos7/DTK-23.04.1-rc4-centos ...@@ -16,12 +16,14 @@ wget http://10.0.50.210:8000/jenkins/rocm/23.04.1/centos7/DTK-23.04.1-rc4-centos
tar -zxvf DTK-23.04.1-rc4-centos7-x86_64.tar.gz tar -zxvf DTK-23.04.1-rc4-centos7-x86_64.tar.gz
source /opt/dtk-23.04.1/env.sh source /opt/dtk-23.04.1/env.sh
``` ```
## 推理
下载模型 (https://models.silero.ai/models/en/en_v5.onnx) 到当前目录,建立wavs文件夹添加测试wav文件 下载模型 (https://models.silero.ai/models/en/en_v5.onnx) 到当前目录,建立wavs文件夹添加测试wav文件
### 推理
``` ```
python3 main.py --model_dir="./en_v5.onnx" --wav_dir="./wavs/" --warmup=1 python3 main.py --model_dir="./en_v5.onnx" --wav_dir="./wavs/" --warmup=1
``` ```
## 性能数据 ## result
## 精度
| size | infer time(ms)| infer time after warmup(ms)| | size | infer time(ms)| infer time after warmup(ms)|
| :------: | :------: | :------: | | :------: | :------: | :------: |
| (4,262144) | 7.6759592751041055 | 0.040782203897833824 | | (4,262144) | 7.6759592751041055 | 0.040782203897833824 |
...@@ -31,6 +33,6 @@ python3 main.py --model_dir="./en_v5.onnx" --wav_dir="./wavs/" --warmup=1 ...@@ -31,6 +33,6 @@ python3 main.py --model_dir="./en_v5.onnx" --wav_dir="./wavs/" --warmup=1
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/asr_onnx https://developer.hpccube.com/codes/modelzoo/asr_onnx
## 参考 ## 参考资料
* [silero-models](https://github.com/snakers4/silero-models) * [silero-models](https://github.com/snakers4/silero-models)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment