README.md 1.75 KB
Newer Older
zhangqha's avatar
zhangqha committed
1
2
# AASIST 
## 模型介绍
3
开源的音频反欺骗的模型
zhangqha's avatar
zhangqha committed
4
5
6
7
8

## 模型结构
aasist,参考链接https://github.com/clovaai/aasist, 参考论文ASIST: Audio Anti-Spoofing using Integrated Spectro-Temporal Graph Attention Networks

## 数据集
zhangqha's avatar
zhangqha committed
9

10
脚本下载方式:
zhangqha's avatar
update  
zhangqha committed
11
```
zhangqha's avatar
zhangqha committed
12
python ./download_dataset.py
zhangqha's avatar
update  
zhangqha committed
13
```
14
手动下载方式:
zhangqha's avatar
zhangqha committed
15

16
ASVspoof2019 dataset: https://datashare.ed.ac.uk/handle/10283/3336链接进行下载LA.zip文件
zhangqha's avatar
zhangqha committed
17

zhangqha's avatar
zhangqha committed
18
19
20
21
22
23
下载后unzip解压

## 推理

### 环境配置
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练的docker镜像:
zhangqha's avatar
zhangqha committed
24
* 推理镜像:
zhangqha's avatar
update  
zhangqha committed
25
```
26
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:aasist-main
zhangqha's avatar
update  
zhangqha committed
27
```
zhangqha's avatar
zhangqha committed
28
激活镜像环境:
zhangqha's avatar
update  
zhangqha committed
29
```
zhangqha's avatar
zhangqha committed
30
31
source /root/env_disc.sh
cd /root/aasist;sh run.sh
zhangqha's avatar
update  
zhangqha committed
32
```
33
* python依赖安装:
zhangqha's avatar
update  
zhangqha committed
34
```
zhangqha's avatar
zhangqha committed
35
pip3 install -r requirement.txt
zhangqha's avatar
update  
zhangqha committed
36
```
zhangqha's avatar
zhangqha committed
37
38
39
### 推理命令,使用BladeDISC进行加速

To evaluate AASIST [1]:
zhangqha's avatar
update  
zhangqha committed
40
```
zhangqha's avatar
zhangqha committed
41
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_"
zhangqha's avatar
zhangqha committed
42

zhangqha's avatar
zhangqha committed
43
44
python3 main.py --eval --config ./config/AASIST.conf
python3 main_opt.py --eval --config ./config/AASIST.conf
zhangqha's avatar
update  
zhangqha committed
45
```
zhangqha's avatar
zhangqha committed
46
47

To evaluate AASIST-L [1]:
zhangqha's avatar
update  
zhangqha committed
48
```
zhangqha's avatar
zhangqha committed
49
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_"
zhangqha's avatar
zhangqha committed
50

zhangqha's avatar
zhangqha committed
51
52
python3 main.py --eval --config ./config/AASIST-L.conf
python3 main_opt.py --eval --config ./config/AASIST-L.conf
zhangqha's avatar
update  
zhangqha committed
53
```
zhangqha's avatar
zhangqha committed
54
55
### 单卡测试
测试命令:
zhangqha's avatar
update  
zhangqha committed
56
```
zhangqha's avatar
zhangqha committed
57
bash run.sh
zhangqha's avatar
update  
zhangqha committed
58
```
59
### 准确率验证
zhangqha's avatar
zhangqha committed
60

61
62
63
准确率验证通过对比优化前与优化后的输出结果,精度差异较小,判定准确率为正常的

## 历史版本
zhangqha's avatar
zhangqha committed
64
* https://developer.hpccube.com/codes/modelzoo/bladedisc_aasist
zhangqha's avatar
zhangqha committed
65
66
67

## 参考
* https://github.com/clovaai/aasist.git