Very simple implementation of GANs, DCGANs, CGANs, WGANs, and etc. with PyTorch for various dataset (MNIST, CARS, CelebA).
You can run the code at Jupyter Notebook. And actually you can also run these codes by using Google Colab immediately (needed downloading some dataset)!
Sometimes ipynb files do not work in Github, please clone and run it in your server.
- python 3.6 (Anaconda)
- pytorch 1.0.0 (updated from 0.4.0. If you want to use the previous version, then find previous commit.)
- you can download the dataset from https://ai.stanford.edu/~jkrause/cars/car_dataset.html
- But I think the dataset is small and needs preprocessing a lot.
- You can also see the samples at ipynbs.
- After DCGAN, DCGAN with the condition is a base model.
- Trained 30 epochs respectively.
- https://colab.research.google.com
- You can run it with GPU(K80) Runtime mode
- Training Vanilla GAN takes only 10 minutes.