Code for reproducing experiments in "Logo Generation and Manipulation with Clustered Generative Adversarial Networks". The models are mainly meant to work with data in HDF5 format (using h5py) such as our Large Logo Dataset, but can easily be adapted to different input data formats (the WGAN models already accepts CIFAR and MNIST).
This repository consists of two main parts:
Our adaptation of DCGAN implementing layer conditioning for training with cluster labels. This is largely based on "DCGAN in Tensorflow"
A modified and extended version of the official TensorFlow code from "Improved Training of Wasserstein GANs".
For usage instructions and prerequisits, please refer to the individual readme's.
A note on the vector.py file: Both versions contain the code for the models themselves as well as a file called vector.py, which is meant to facilitate experimentation with a trained model, such as sampling it and performimg interpolations and vector arithmetic in latent space. At this time the two implementations differ slightly to integrate with the DCGAN and WGAN models. If there is sufficient interest, we might also make this a independent module that can be used with any GAN (or VaE) model that provides some specified interface. We believe this would save the research community a significant amount of work if its usage would spread. Please feel free to contact us should you be interesed in using and/or help us develop such a module.
DCGAN - LLD-icon with 100 AE clusters
WGAN - LLD-icon with 128 RC clusters
WGAN - LLD-icon-sharp with 128 RC clusters
WGAN - LLD-icon-sharp with 16 RC clusters