Counting 3,742 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Recommendation

  • Our GAN based work for facial attribute editing - AttGAN.

New

  • We re-implement these GANs by Tensorflow 2! The old version is here: v1 or in the "v1" directory.


GANs - Tensorflow 2

Tensorflow 2 implementations of DCGAN, LSGAN, WGAN-GP and DRAGAN.

Exemplar results

Fashion-MNIST

DCGAN LSGAN WGAN-GP DRAGAN

CelebA

DCGAN LSGAN
WGAN-GP DRAGAN

Anime

WGAN-GP DRAGAN

Usage

  • Prerequisites

    • Tensorflow 2.0 Alpha pip install tensorflow==2.0.0-alpha0
    • Tensorflow Addons pip install tensorflow-addons
    • (if you meet "tf.summary.histogram fails with TypeError" pip install --upgrade tb-nightly)
    • scikit-image, oyaml, tqdm
    • Python 3.6
  • Datasets

  • Examples of training

    • Fashion-MNIST DCGAN

      CUDA_VISIBLE_DEVICES=0 python train.py --dataset=fashion_mnist --epoch=25 --adversarial_loss_mode=gan
    • CelebA DRAGAN

      CUDA_VISIBLE_DEVICES=0 python train.py --dataset=celeba --epoch=25 --adversarial_loss_mode=gan --gradient_penalty_mode=dragan
    • Anime WGAN-GP

      CUDA_VISIBLE_DEVICES=0 python train.py --dataset=anime --epoch=200 --adversarial_loss_mode=wgan --gradient_penalty_mode=wgan-gp --n_d=5
    • see more training exampls in commands.sh

    • tensorboard for loss visualization

      tensorboard --logdir ./output/fashion_mnist_gan/summaries --port 6006

Latest Releases
v1
 Apr. 6 2019
v1
 Feb. 25 2019