Home

Awesome

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

Datasets

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

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.