/GAN-Tutorial

Simple Implementation of many GAN models with PyTorch.

Primary LanguageJupyter Notebook

GANs Tutorial

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.

Requirements

  • 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.)

Implementation List

MNIST

CARS (Stanford dataset)

CelebA (aligned dataset)

Experimental Results

  • You can also see the samples at ipynbs.
  • After DCGAN, DCGAN with condition is a base model.
  • Trained 30 epochs respectively.

Vanilla GAN

Conditional GAN

DC GAN

WGAN-gp

infoGAN w/ walking code 1

infoGAN w/ walking code 2

BEGAN random samples (20 epochs)

BEGAN interpolation

GAN with R1 regularization random samples (20 epochs)

GAN with R1 regularization interpolation

Colab