/DeepFool_MNIST

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

Primary LanguageJupyter Notebook

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