pbaljeka / progressive_growing_of_gans Goto Github PK

Progressive Growing of GANs for Improved Quality, Stability, and Variation

Tero Karras (NVIDIA), Timo Aila (NVIDIA), Samuli Laine (NVIDIA), Jaakko Lehtinen (NVIDIA and Aalto University)

We describe a new training methodology for generative adversarial networks. The key idea is to grow both the generator and discriminator progressively, starting from low-resolution images, and add new layers that deal with higher resolution details as the training progresses. This greatly stabilizes the training and allows us to produce images of unprecedented quality, e.g., CelebA images at 1024² resolution. We also propose a simple way to increase the variation in generated images, and achieve a record inception score of 8.80 in unsupervised CIFAR10. Additionally, we describe several small implementation details that are important for discouraging unhealthy competition between the generator and discriminator. Finally, we suggest a new metric for evaluating GAN results, both in terms of image quality and variation. As an additional contribution we construct a higher quality version of the CelebA dataset that allows meaningful exploration up to the resolution of 1024² pixels.

Links

• Paper (research.nvidia.com)
• Video (YouTube)
• Pre-trained networks + 100,000 generated PNGs for each dataset (Google Drive)
• Datasets (currently unavailable)

License

The source code is available under the CC BY-NC license:

# Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.
#
# This work is licensed under the Creative Commons Attribution-NonCommercial
# 4.0 International License. To view a copy of this license, visit
# http://creativecommons.org/licenses/by-nc/4.0/ or send a letter to
# Creative Commons, PO Box 1866, Mountain View, CA 94042, USA.

Compatibility

We have tested the implementation using the following system configuration:

• NVIDIA DGX-1 with Tesla P100
• BaseOS 2.1.0, 4.4.0-92-generic kernel
• NVIDIA driver 384.81, CUDA Toolkit 9.0
• Bleeding-edge version of Theano and Lasagne from Oct 17, 2017

We are planning to add TensorFlow support in the near future.