Skip to content

mbinkowski/MMD-GAN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Official Tensorflow implementation for reproducing the results of Demystifying MMD GANs.

The repository contains code for reproducing experiments of uncoditional image generation with MMD GANs and other benchmark GAN models.

If you're only interested in the new KID metric, check out compute_scores.py.

References

Mikołaj Bińkowski, Dougal J. Sutherland, Michael N. Arbel and Athur Gretton. Demystifying MMD GANs. ICLR 2018 (openreview; poster).

Model features

Requirements

  • python >= 3.6
  • tensorflow-gpu >= 1.3
  • PIL, lmdb, numpy, matplotlib
  • machine with GPU(s). At least 2 GPUs are needed for experiments with Celeb-A dataset.

Datasets

The code works with several common datasets with different resolutions. The experiments include

  • 28x28 MNIST,
  • 32x32 Cifar10,
  • 64x64 LSUN Bedrooms,
  • 160x160 Celeb-A.

LSUN, MNIST and Celeb-A datasets can be downloaded using the script.

Benchmarks

We compare MMD GANs with WGAN-GP and Cramer GAN.

Running the code

Each of the following scripts launches the training of MMD GAN on respective dataset: mnist.sh, cifar10.sh, lsun.sh, celeba.sh. To train the benchmark models, change the variable $MODEL to WGAN or CRAMER. To train all three models set $MODEL=ALL.

Feel free to contact Mikołaj Bińkowski (mikbinkowski at gmail.com) with any questions and issues.

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.0%
  • Shell 4.0%