DeepFool_MNIST
DeepFool_MNIST copied to clipboard
Based on Pytorch, the Adversarial Attack algorithm DeepFool, targeting the Mnist data set and ResNet18 network
基于Mnist,DeepFool攻击ResNet18
文件说明
data:Mnist数据集images:对抗样本,9957张,剩余43张图像网络本身无法正确分类,为原图像model:ResNet18的训练函数,和预训练参数文件logs:训练日志
代码文档说明
model.py:ResNet18网络定义utils.py:其他函数deepfool_fashion.py:DeepFool核心算法,当使用其他数据集时,代码中的图像预处理需要微调Resnet-Deepfool.ipynd:主函数tran.py:模型训练函数。
环境配置
- python 3.7
- pytorch 1.4.0
- cuda 10.1