/CE-GAN

Primary LanguagePython

CE-GAN

Prerequisites

  • Linux or macOS
  • Python 3
  • CPU or NVIDIA GPU + CUDA CuDNN

Getting Started

Preparation

  • Preparing .npz files for Pytorch Inception metrics evaluation (cifar10 as an example):
python envs_repo/inception_pytorch/calculate_inception_moments.py --dataset C10 --data_root datasets
  • Preparing .tgz files for Tensorflow Inception metrics evaluation
  • Preparing .npz files for Tensorflow FID metrics evaluation

Operator GANs Training

An example of GAN training command was saved in ./scripts/operator_cifar10.sh. Train a model (cifar10 as an example):

bash ./scripts/operator_cifar10.sh

CE-GAN Training

An example of CE-GAN training command was saved in ./scripts/cegan_cifar10.sh. Train a model (cifar10 as an example):

bash ./scripts/cegan_cifar10.sh

Acknowledgments

Pytorch Inception metrics code from BigGAN-PyTorch.

TensorFlow Inception Score code from OpenAI's Improved-GAN..

TensorFlow FID code from TTUR.