owruby / APE-GAN

Pytorch Implementation of APE-GAN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

APE-GAN

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

MNIST

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

CIFAR10

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

About

Pytorch Implementation of APE-GAN


Languages

Language:Python 100.0%