README.md 1.45 KB
Newer Older
chenxj's avatar
chenxj committed
1
2
3
# asr_onnx
## 论文

chenxj's avatar
chenxj committed
4
## 模型结构
chenxj's avatar
chenxj committed
5
![image](https://developer.hpccube.com/codes/modelzoo/asr_onnx/-/raw/main/resources/silero_stt_model.jpg)
dcuai's avatar
dcuai committed
6
## 算法原理
chenxj's avatar
chenxj committed
7
8
9
10
![image](https://developer.hpccube.com/codes/modelzoo/asr_onnx/-/raw/main/resources/asr.png)
## 数据集

## 环境配置
chenxj's avatar
chenxj committed
11
12
13
[光源](https://www.sourcefind.cn/#/service-details)可拉取推理的docker镜像,在[光合开发者社区](https://cancon.hpccube.com:65024/4/main/)可下载onnxruntime安装包。asr_onnx推荐的镜像如下:
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest
chenxj's avatar
chenxj committed
14
15
16
docker run -d -t --privileged --device=/dev/kfd --device=/dev/dri/ --network=host --group-add video --name asr-test image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest
docker exec -it asr-test bash
pip3 install onnxruntime.whl
chenxj's avatar
chenxj committed
17
```
chenxj's avatar
chenxj committed
18
下载模型 (https://models.silero.ai/models/en/en_v5.onnx) 到当前目录,建立wavs文件夹添加测试wav文件
chenxj's avatar
chenxj committed
19
## 推理
chenxj's avatar
chenxj committed
20
21
22
```
python3 main.py --model_dir="./en_v5.onnx" --wav_dir="./wavs/" --warmup=1
```
dcuai's avatar
dcuai committed
23
## result
chenxj's avatar
chenxj committed
24
25
26
27
28
29
30
31
![image](https://developer.hpccube.com/codes/modelzoo/asr_onnx/-/raw/main/resources/asr_result.png)
### 精度
暂无
## 应用场景
### 算法类别
语音识别
### 热点应用行业
交通,金融,医疗,教育,家居
chenxj's avatar
chenxj committed
32
33
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/asr_onnx
dcuai's avatar
dcuai committed
34
## 参考资料
chenxj's avatar
chenxj committed
35
* [silero-models](https://github.com/snakers4/silero-models)
sugon_cxj's avatar
sugon_cxj committed
36