Skip to content

Implementation of the "Improved Training of Wasserstein GANs" paper in TensorFlow

Notifications You must be signed in to change notification settings

bgavran/Improved_WGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Improved training of Wasserstein GANs


In this project, the paper Improved training of Wasserstein GANs was implemented in Tensorflow 1.2.0 and Python 3.6.

The paper is the improvement of the Wasserstein GAN paper, which again is the improvement over the original Generative Adversarial Networks paper.

Each of those extension papers represents a step to a more stable training regime.

Improved WGAN, compared to GAN:

  • Uses a different distance measure to compare distributions (Wasserstein instead of KL-divergence)
  • Enforces the Lipschitz constraint on the critic using gradient penalty

The images on the top are some of the best results with DCGAN and custom upsampling architecture. The dataset used was Labeled Faces in the Wild, the deep-funneled version.

Sample critic and generator training provided below:

This project was created as a part of the FER course Analysis of massive data sets.

About

Implementation of the "Improved Training of Wasserstein GANs" paper in TensorFlow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages