shariqiqbal2810 / WGAN-GP-PyTorch

Implementation of "Improved Training of Wasserstein GANs" in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

WGAN-GP-PyTorch

Implementation of Improved Training of Wasserstein GANs (Gulrajani, et al.) in PyTorch

Requires

  • PyTorch
  • Numpy
  • scipy (for loading SVHN .mat file)

Training

Run the following code to train on the SVHN dataset

python code/train_SVHN.py <location-of-SVHN-mat-file> <model-name>

The parameters can be changed through the command line. Use the --help flag for more details

Results

Generated vs Real Data This image compares real data and data generated from the model Image Interpolation This image demonstrates an interpolation from two points in the noise space of the generator. A straight line was drawn between two randomly drawn vectors and the points along that line were fed into the generator to produce these images.

About

Implementation of "Improved Training of Wasserstein GANs" in PyTorch

License:MIT License


Languages

Language:Jupyter Notebook 93.7%Language:Python 6.3%