TF-GAN is a lightweight library for training and evaluating Generative Adversarial Networks (GANs).
- Can be installed with
pip
usingpip install tensorflow-gan
, and used withimport tensorflow_gan as tfgan
- Well-tested examples
- Interactive introduction to TF-GAN in colaboratory
TF-GAN is composed of several parts, which are designed to exist independently:
- Core: the main infrastructure needed to train a GAN. Set up training with any combination of TF-GAN library calls, custom-code, native TF code, and other frameworks
- Features: common GAN operations and normalization techniques, such as instance normalization and conditioning.
- Losses: losses and penalties, such as the Wasserstein loss, gradient penalty, mutual information penalty, etc.
- Evaluation:
standard GAN evaluation metrics.
Use
Inception Score
,Frechet Distance
, orKernel Distance
with a pretrained Inception network to evaluate your unconditional generative model. You can also use your own pretrained classifier for more specific performance numbers, or use other methods for evaluating conditional generative models. - Examples: simple examples on how to use TF-GAN, and more complicated state-of-the-art examples
Numerous projects inside Google. The following are some published papers that use TF-GAN:
- Are GANs Created Equal? A Large-Scale Study
- The GAN Landscape: Losses, Architectures, Regularization, and Normalization
- Self-Attention Generative Adversarial Networks
- Large Scale GAN Training for High Fidelity Natural Image Synthesis
- Discriminator rejection sampling
- The GAN Landscape: Losses, Architectures, Regularization, and Normalization
- GANSynth: Adversarial Neural Audio Synthesis
Training in TF-GAN typically consists of the following steps:
- Specify the input to your networks.
- Set up your generator and discriminator using a
GANModel
. - Specify your loss using a
GANLoss
. - Create your train ops using a
GANTrainOps
. - Run your train ops.
At each stage, you can either use TF-GAN's convenience functions, or you can perform the step manually for fine-grained control.
There are various types of GAN setup. For instance, you can train a generator to sample unconditionally from a learned distribution, or you can condition on extra information such as a class label. TF-GAN is compatible with many setups, and we demonstrate in the well-tested examples directory
- Paolo Galeone, [email protected], github
- (Documentation) David Westbrook, [email protected]
- Joel Shor, [email protected], github
- Aaron Sarna, [email protected], github
- Joel Shor, [email protected], github