Dawn-David / DeepFool_MNIST Public

Notifications You must be signed in to change notification settings
Fork 2
Star 16

Based on Pytorch, the Adversarial Attack algorithm DeepFool, targeting the Mnist data set and ResNet18 network

16 stars 2 forks Branches Tags Activity

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
model		model
.gitignore		.gitignore
Resnet-Deepfool.ipynb		Resnet-Deepfool.ipynb
deepfool_fashion.py		deepfool_fashion.py
readme.md		readme.md
train.py		train.py

Repository files navigation

基于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