Generative-models
Generative models using Keras and Pytorch
Requirement
- python3
- keras or pytorch
Keras
VAE
- parser description
- -d : dataset name(mnist or fashion_mnist)
- --latentdim : latent dimention(default=2)
- --epoch : the number of epochs(default=2)
- -b : the number of batch size(default=16)
Quick start
python3 ./VAE.py -d [dataset_name]
GAN
- parser description
- -d : dataset name(mnist, fashio_manist or cifar10)
- -b : batch size(default=128)
- --epoch : the number of epochs(default=20)
- --latentdim : latent dimention(default=100)
- --leakyrelu : use leeaky-ReLU or not(default=True)
- --tilt : tilt of leaky-ReLU(default=0.2)
- --droprate : dropout rate(default=0.3)
- --genoptim : generator's optimizer(default=adam)
- --discoptim : discriminator's optimizer(default=adam)
- --softlabel : use softlabel or not(default=False)
Quick start
python3 ./train.py -d [dataset_name]