DRAGAN Implementation with Auxiliary Classifier based on a modification from SRResNet
Attempt at recreating results from: Towards the Automatic Anime Characters Creation with Generative Adversarial Networks (https://arxiv.org/pdf/1708.05509.pdf)
Experimented with the CelebA Dataset.
TODO:
- Add results on tagged anime dataset
- Double check training procedure
- Add Interpolation Results
Pretrained model available at https://drive.google.com/open?id=1JTpjlfVO-Frjr9_rdytxiuQZU7ixCF53
Download Cropped, Aligned CelabA Dataset at http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html
Code is based on the following repositories:
- https://github.com/ai-tor/PyTorch-SRGAN
- https://github.com/jfsantos/dragan-pytorch
- https://github.com/pytorch/examples/blob/master/imagenet/main.py
To run the code, adjust settings.py
and call python3 main.py
.
Epoch 0-18: