Commit 40515c7e authored by zhangqha's avatar zhangqha
Browse files

update

parent c2947e5e
...@@ -8,8 +8,9 @@ aasist,参考链接https://github.com/clovaai/aasist, 参考论文ASIST: Audio ...@@ -8,8 +8,9 @@ aasist,参考链接https://github.com/clovaai/aasist, 参考论文ASIST: Audio
## 数据集 ## 数据集
脚本下载方式: 脚本下载方式:
```
python ./download_dataset.py python ./download_dataset.py
```
手动下载方式: 手动下载方式:
ASVspoof2019 dataset: https://datashare.ed.ac.uk/handle/10283/3336链接进行下载LA.zip文件 ASVspoof2019 dataset: https://datashare.ed.ac.uk/handle/10283/3336链接进行下载LA.zip文件
...@@ -21,39 +22,40 @@ ASVspoof2019 dataset: https://datashare.ed.ac.uk/handle/10283/3336链接进行 ...@@ -21,39 +22,40 @@ ASVspoof2019 dataset: https://datashare.ed.ac.uk/handle/10283/3336链接进行
### 环境配置 ### 环境配置
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练的docker镜像: 提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练的docker镜像:
* 推理镜像: * 推理镜像:
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:aasist-main docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:aasist-main
```
激活镜像环境: 激活镜像环境:
```
source /root/env_disc.sh source /root/env_disc.sh
cd /root/aasist;sh run.sh cd /root/aasist;sh run.sh
```
* python依赖安装: * python依赖安装:
```
pip3 install -r requirement.txt pip3 install -r requirement.txt
```
### 推理命令,使用BladeDISC进行加速 ### 推理命令,使用BladeDISC进行加速
To evaluate AASIST [1]: To evaluate AASIST [1]:
```
export TORCH_MHLO_OP_WHITE_LIST="aten::max;aten::batch_norm;aten::abs,aten::selu;prim::NumToTensor;aten::zeros_like;aten::size;aten::narrow;aten::cat;aten::selu_" export TORCH_MHLO_OP_WHITE_LIST="aten::max;aten::batch_norm;aten::abs,aten::selu;prim::NumToTensor;aten::zeros_like;aten::size;aten::narrow;aten::cat;aten::selu_"
python3 main.py --eval --config ./config/AASIST.conf python3 main.py --eval --config ./config/AASIST.conf
python3 main_opt.py --eval --config ./config/AASIST.conf python3 main_opt.py --eval --config ./config/AASIST.conf
```
To evaluate AASIST-L [1]: To evaluate AASIST-L [1]:
```
export TORCH_MHLO_OP_WHITE_LIST="aten::max;aten::batch_norm;aten::abs,aten::selu;prim::NumToTensor;aten::zeros_like;aten::size;aten::narrow;aten::cat;aten::selu_" export TORCH_MHLO_OP_WHITE_LIST="aten::max;aten::batch_norm;aten::abs,aten::selu;prim::NumToTensor;aten::zeros_like;aten::size;aten::narrow;aten::cat;aten::selu_"
python3 main.py --eval --config ./config/AASIST-L.conf python3 main.py --eval --config ./config/AASIST-L.conf
python3 main_opt.py --eval --config ./config/AASIST-L.conf python3 main_opt.py --eval --config ./config/AASIST-L.conf
```
### 单卡测试 ### 单卡测试
测试命令: 测试命令:
```
bash run.sh bash run.sh
```
### 准确率验证 ### 准确率验证
准确率验证通过对比优化前与优化后的输出结果,精度差异较小,判定准确率为正常的 准确率验证通过对比优化前与优化后的输出结果,精度差异较小,判定准确率为正常的
......
# 模型名称 # 模型名称
modelName=aasist modelName=bladedisc_aasist
# 模型描述 # 模型描述
modelDescription=aasist是开源的音频反欺骗的模型 modelDescription=bladedisc_aasist是开源的音频反欺骗的模型
# 应用场景(多个标签以英文逗号分割) # 应用场景(多个标签以英文逗号分割)
appScenario=NLP,语音模型,推理 appScenario=NLP,语音模型,推理
# 框架类型(多个标签以英文逗号分割) # 框架类型(多个标签以英文逗号分割)
......
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