APE-GAN

Implementation APE-GAN (https://arxiv.org/pdf/1707.05474.pdf)

MNIST

1. Train CNN and Generate Adversarial Examples(FGSM)

python generate.py --eps 0.15

2. Train APE-GAN

python train.py --checkpoint ./checkpoint/mnist

3. Test

python test_model.py --eps 0.15 --gan_path ./checkpoint/mnist/3.tar

CIFAR-10

1. Train CNN and Generate Adversarial Examples(FGSM)

python generate.py --data cifar --eps 0.01

2. Train APE-GAN

python train.py --data cifar --epochs 30 --checkpoint ./checkpoint/cifar

3. Test

python test_model.py --data cifar --eps 0.01 --gan_path ./checkpoint/cifar/10.tar