igul222 / improved_wgan_training

Code for reproducing experiments in "Improved Training of Wasserstein GANs"
MIT License
2.35k stars 668 forks source link

Inference of class information in the improved wasserstein GANs #71

Closed kristosh closed 6 years ago

kristosh commented 6 years ago

I am using the implementation of improved wassterstein gans in order to generate images from my database (a database with faces). However, in the current implementation of the gans there is not an inference of the class information in the code. What should i do in order to change the code by taking into account also the class information?