text-to-image
text-to-image copied to clipboard
Text to Image Synthesis using Generative Adversarial Networks
Text to Image Synthesis using Generative Adversarial Networks
This is the official code for Text to Image Synthesis using Generative Adversarial Networks. Please be aware that the code is in an experimental stage and it might require some small tweaks.
If you find my research useful, please use the following to cite:
@article{Bodnar2018TextTI,
title={Text to Image Synthesis Using Generative Adversarial Networks},
author={Cristian Bodnar},
journal={CoRR},
year={2018},
volume={abs/1805.00676}
}
Images generated by the Conditional Wasserstein GAN
As it can be seen, the generated images do not suffer from mode collapse.
Illustration of Conditional Wasserstein Progressive Growing GAN on the flowers dataset:
Samples from the birds dataset
How to download the dataset
- Setup your
PYTHONPATH
to point to the root directory of the project. - Download the preprocessed flowers text descriptions
and extract them in the
/data
directory. - Download the images from Oxford102
and extract the images in
/data/flowers/jpg
. You can alternatively runpython preprocess/download_flowers_dataset.py
from the root directory of the project. - Run the
python preprocess/preprocess_flowers.py
script from the root directory of the project.
Requirements
- python 3.6
- tensorflow 1.4
- scipy
- numpy
- pillow
- easydict
- imageio
- pyyaml