AlainConnor / gan_practice

dcgan,wgan and improved wgan implementation by tensorflow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gan_practice

simple dcgan,wgan and improved wgan implementation by tensorflow, this is for practice purpose. tensorflow version=1.1.0

data preparation

download MNIST data from http://yann.lecun.com/exdb/mnist/ and unzip data to some dir,for example

data
|__ 10k-images-idx3-ubyte
|__ train-images-idx3-ubyte

training

python dcgan.py or wgan.py or wgan_gp.py \
                --data_path=data \
                --mode=train \
                --model_dir=models \
                --batch_size=100 \
                --epoch_num=100 \
                --rand_dim=128

generating

python dcgan.py or wgan.py or wgan_gp.py \
                --data_path=data \
                --mode=infer \
                --model_dir=models \
                --batch_size=100 \
                --epoch_num=100 \
                --rand_dim=128

some results

  • dcgan 100 epoch results

* wgan 100 epoch results

* improved wgan 100 epoch results

reference

ezoic increase your site revenue

About

dcgan,wgan and improved wgan implementation by tensorflow

License:Apache License 2.0


Languages

Language:Python 100.0%