Awesome Open Source
Awesome Open Source

News

  • 28 June 2019: We re-implement these GANs by Pytorch 1.1! The old version is here: v0 or in the "v0" directory.
  • Tensorflow 2 Version


GANs - Pytorch

Pytorch implementations of DCGAN, LSGAN, WGAN-GP(LP) and DRAGAN.

Exemplar results

Fashion-MNIST

DCGAN LSGAN WGAN-GP DRAGAN

CelebA

DCGAN LSGAN
WGAN-GP DRAGAN
WGAN-LP DRAGAN-LP

Anime

WGAN-GP DRAGAN

Usage

  • Prerequisites

    • PyTorch 1.1
    • tensorboardX
    • 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=1-gp --gradient_penalty_sample_mode=dragan
      
    • Anime WGAN-GP

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

    • tensorboard for loss visualization

      tensorboard --logdir ./output/fashion_mnist_gan/summaries --port 6006
      
Alternatives To Dcgan Lsgan Wgan Gp Dragan Pytorch
Select To Compare


Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (892,476
Pytorch (22,677
Mnist (4,497
Anime (3,129
Dcgan (819
Celeba (587
Wgan (406
Wgan Gp (128
Lsgan (105
Dragan (59