J14032016 / Improved_WGAN

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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


Languages

Language:Python 100.0%