GAN-Experiment

News

An anime branch for anime generation is now AVAILABLE!!! It was developed by my friend hsj.

Usage

python main.py --checkpoint_dir checkpoint --model dcgan --dataset MNIST --device 0

TO DO

• Add support for CelebA and LSUN dataset.
• Tensorboard to visualize the training process and generated images
• Multi-GPU training

Reference

• A good tutorial for GAN beginners.
• The calculation of Inception Score is borrowed from this repository.
• The calculation of FID is borrowed from this repository.
• The implementation of self-attention is borrowed from this repository, and the download script for CelebA and LSUN download.sh is also borrowed from it.