Skip to content

bgavran/Improved_WGAN

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
src
 
 
 
 
 
 

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