朱瑞 91baeb0f58 | ||
---|---|---|
clean_resized_images | ||
torch_nets | ||
Normalize.py | ||
README.md | ||
attack.py | ||
dataset.py | ||
evaluate.py | ||
evaluate_cnn.py | ||
methods.py | ||
model.py | ||
run_evaluate.sh | ||
utils.py |
README.md
Requirements
- Python 3.6.13
- Pytorch 1.7.1
- Torchvision 0.8.2
- Numpy 1.19.2
- Pillow 8.3.1
- Timm 0.4.12
- Scipy 1.5.4
生成对抗样本
python attack.py --attack ADG --batch_size 1 --model_name vit_base_patch16_224
评估成功率
bash run_evaluate.sh model_vit_base_patch16_224-method python evaluate_cnn.py