yejg2017 / stackgan-text2image

text to image learning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

stackgan-text2image

text to image learning

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

Requirements

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

thanks to

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

License

Apache 2.0

About

text to image learning


Languages

Language:Python 71.3%Language:Jupyter Notebook 28.6%Language:Shell 0.1%