cycle-image-gan
Based on https://github.com/taoxugit/AttnGAN/tree/master/code
Paper https://arxiv.org/abs/2003.12137
Spring 2019 CS 224U Project
- BERT encoder
- Cycle-GAN
- Image2Text encoder
Download Data
- Download AttnGAN preprocessed data and captions birds
- Download the birds image data. Extract them to
data/birds/
Instructions
- pretrain STREAM
python pretrain_STREAM.py --cfg cfg/STREAM/bird.yaml --gpu 0
- train CycleGAN
python main.py --cfg cfg/bird_cycle.yaml --gpu 0