cwgan-gp

cwgan-gp copied to clipboard

Conditional WGAN-GP (cWGAN-GP) in Keras

Conditional version of WGAN-GP is the combination of cgan and wgan-gp. It can be used to generate samples of a particular class.

Datasets

Mnist dataset is used in the experiment.

Usage

python3 cwgan_gp.py

Result

Generate image of a specific class

Use the code wgan.generate_images(class_names).

Defining a new task for cWGAN-GP

Need to change img_rows, img_cols, channels, nclasses, load your own dataset and change the network structure to corresponding size.