Collection of all GANs

The original code address

https://github.com/hwalsuklee/tensorflow-generative-model-collections

tensorflow-generative-model-collections

Tensorflow implementation of various GANs and VAEs.

Pytorch version

Pytorch Version is now availabel at https://github.com/znxlwm/pytorch-generative-model-collections

Project progress

It's adapted to the cifar10, celebA. Details can be reached via email.

Generative Adversarial Networks (GANs)

Lists

NamePaper LinkValue Function
GANArxiv
LSGANArxiv
WGANArxiv
WGAN-GPArxiv
DRAGANArxiv
CGANArxiv
infoGANArxiv
ACGANArxiv
EBGANArxiv
BEGANArxiv
VAE_GANArxivto be add
Variants of GAN structure

Results for mnist

Network architecture of generator and discriminator is the exaclty sames as in infoGAN paper.
For fair comparison of core ideas in all gan variants, all implementations for network architecture are kept same except EBGAN and BEGAN. Small modification is made for EBGAN/BEGAN, since those adopt auto-encoder strucutre for discriminator. But I tried to keep the capacity of discirminator.

The following results can be reproduced with command:

python main.py --dataset mnist --gan_type <TYPE> --epoch 25 --batch_size 64
Random generation

All results are randomly sampled.

NameEpoch 2Epoch 10Epoch 25
GAN
LSGAN
WGAN
WGAN-GP
DRAGAN
EBGAN
BEGAN
VAE_GAN
Conditional generation

Each row has the same noise vector and each column has the same label condition.

NameEpoch 1Epoch 10Epoch 25
CGAN
ACGAN
infoGAN
InfoGAN : Manipulating two continous codes

Results for fashion-mnist

Comments on network architecture in mnist are also applied to here.
Fashion-mnist is a recently proposed dataset consisting of a training set of 60,000 examples and a test set of 10,000 examples. Each example is a 28x28 grayscale image, associated with a label from 10 classes. (T-shirt/top, Trouser, Pullover, Dress, Coat, Sandal, Shirt, Sneaker, Bag, Ankle boot)

The following results can be reproduced with command:

python main.py --dataset fashion-mnist --gan_type <TYPE> --epoch 40 --batch_size 64
Random generation

All results are randomly sampled.

NameEpoch 1Epoch 20Epoch 40
GAN
LSGAN
WGAN
WGAN-GP
DRAGAN
EBGAN
BEGAN
VAE_GANto be add  
Conditional generation

Each row has the same noise vector and each column has the same label condition.

NameEpoch 1Epoch 20Epoch 40
CGAN
ACGAN
infoGAN

Without hyper-parameter tuning from mnist-version, ACGAN/infoGAN does not work well as compared with CGAN.
ACGAN tends to fall into mode-collapse.
infoGAN tends to ignore noise-vector. It results in that various style within the same class can not be represented.

InfoGAN : Manipulating two continous codes

Some results for celebA or cifar10

(to be added)

Variational Auto-Encoders (VAEs)

Lists

NamePaper LinkLoss Function
VAEArxiv
CVAEArxiv
DVAEArxiv(to be added)
AAEArxiv(to be added)
Variants of VAE structure

Results for mnist

Network architecture of decoder(generator) and encoder(discriminator) is the exaclty sames as in infoGAN paper. The number of output nodes in encoder is different. (2x z_dim for VAE, 1 for GAN)

The following results can be reproduced with command:

python main.py --dataset mnist --gan_type <TYPE> --epoch 25 --batch_size 64
Random generation

All results are randomly sampled.

NameEpoch 1Epoch 10Epoch 25
VAE
GAN

Results of GAN is also given to compare images generated from VAE and GAN. The main difference (VAE generates smooth and blurry images, otherwise GAN generates sharp and artifact images) is cleary observed from the results.

Conditional generation

Each row has the same noise vector and each column has the same label condition.

NameEpoch 1Epoch 10Epoch 25
CVAE
CGAN

Results of CGAN is also given to compare images generated from CVAE and CGAN.

Learned manifold

The following results can be reproduced with command:

python main.py --dataset mnist --gan_type VAE --epoch 25 --batch_size 64 --dim_z 2

Please notice that dimension of noise-vector z is 2.

NameEpoch 1Epoch 10Epoch 25
VAE

Results for fashion-mnist

Comments on network architecture in mnist are also applied to here.

The following results can be reproduced with command:

python main.py --dataset fashion-mnist --gan_type <TYPE> --epoch 40 --batch_size 64
Random generation

All results are randomly sampled.

NameEpoch 1Epoch 20Epoch 40
VAE
GAN

Results of GAN is also given to compare images generated from VAE and GAN.

Conditional generation

Each row has the same noise vector and each column has the same label condition.

NameEpoch 1Epoch 20Epoch 40
CVAE
CGAN

Results of CGAN is also given to compare images generated from CVAE and CGAN.

Learned manifold

The following results can be reproduced with command:

python main.py --dataset fashion-mnist --gan_type VAE --epoch 25 --batch_size 64 --dim_z 2

Please notice that dimension of noise-vector z is 2.

NameEpoch 1Epoch 10Epoch 25
VAE

Results for celebA

(We are building, GAN has been adapted)

Folder structure

The following shows basic folder structure.

├── main.py # gateway
├── data
│   ├── mnist # mnist data (not included in this repo)
│   |   ├── t10k-images-idx3-ubyte.gz
│   |   ├── t10k-labels-idx1-ubyte.gz
│   |   ├── train-images-idx3-ubyte.gz
│   |   └── train-labels-idx1-ubyte.gz
│   └── fashion-mnist # fashion-mnist data (not included in this repo)
│       ├── t10k-images-idx3-ubyte.gz
│       ├── t10k-labels-idx1-ubyte.gz
│       ├── train-images-idx3-ubyte.gz
│       └── train-labels-idx1-ubyte.gz
├── GAN.py # vainilla GAN
├── ops.py # some operations on layer
├── utils.py # utils
├── logs # log files for tensorboard to be saved here
└── checkpoint # model files to be saved here

Training Tips

Use SGD for discriminator and ADAM for generator prepare a new model

Acknowledgements

This implementation has been based on this repository and tested with Tensorflow over ver1.0 on Windows 10 and Ubuntu 14.04.

GANs Summary

https://github.com/TwistedW/tensorflow-GANs

https://github.com/TwistedW/tensorflow-GANs
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值