Nothing Special   »   [go: up one dir, main page]

Skip to content

Yangyangii/GAN-Tutorial

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

41 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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