完整说明文档,请参阅 https://aisafety.readthedocs.io/zh_CN/latest/
-
Install pytorch
The code is tested on python 3.6.5 and torch 1.6.0
-
Install requirement
pip install -r requirement.txt
-
Clone the repository
git clone git@github.com:DIG-Beihang/AISafety.git
-
change root to
test
cd test
-
run
testimport.py
python testimport.py
- --attack_method: 用于生成对抗样本的攻击算法
- --evaluation_method:用于执行评测的评测算法
- --Data_path:传入数据集路径
- --...:更多参数说明请参看完整说明文档中介绍