/logo-gen

Accompanying code for the paper "Logo Synthesis and Manipulation with Clustered Generative Adversarial Networks"

Primary LanguagePythonMIT LicenseMIT

Logo Generation and Manipulation with Clustered Generative Adversarial Networks

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:

DCGAN

Our adaptation of DCGAN implementing layer conditioning for training with cluster labels. This is largely based on "DCGAN in Tensorflow"

WGAN

A modified and extended version of the official TensorFlow code from "Improved Training of Wasserstein GANs".

Detailed Instructions

For usage instructions and prerequisits, please refer to the individual readme's.

vector.py

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.

Pretrained Models

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

WGAM - LLD-logo with 16 RC clusters

Large Logo Dataset

Website of our LLD Dataset