yejg2017 / stackgan-text2image

text to image learning

stackgan-text2image

text to image learning

Image Source : Generative Adversarial Text-to-Image Synthesis Paper

Requirements

• TensorFlow 1.0+
• TensorLayer 1.4+
• NLTK : for tokenizer

Datasets

• The model is currently trained on the flowers dataset. Download the images from here .Also download the captions from this link. Extract the archive, copy the text_c10 folder and paste it in 102flowers/text_c10/class_*.

N.B You can downloads all data files needed manually or simply run the downloads.py and put the correct files to the right directories.

python downloads.py

Neccessary folders

mkdir checkpoints cvpr2016_flowers 102flowers Data imagenet

Codes

• downloads.py download Oxford-102 flower dataset and caption files(run this first).
• data_loader.py load data for further processing.
• train_txt2im.py train a text to image model.
• utils.py helper functions.
• model.py models.

References

• Generative Adversarial Text-to-Image Synthesis Paper
• Generative Adversarial Text-to-Image Synthesis Torch Code
• Skip Thought Vectors Paper
• Skip Thought Vectors Code
• Generative Adversarial Text-to-Image Synthesis with Skip Thought Vectors TensorFlow code
• DCGAN in Tensorflow

thanks to

https://github.com/zsdonghao/text-to-image 嘻嘻

License

Apache 2.0