ControlGAN-Tensorflow

ControlGAN-Tensorflow copied to clipboard

ControlGAN — Simple TensorFlow Implementation [Paper]

: Controllable Text-to-Image Generation

Dataset

Text

• birds
• coco

Image

• birds
• coco

Usage

├── dataset
   └── YOUR_DATASET_NAME
       ├── images
           ├── domain1 (domain folder)
               ├── xxx.jpg (domain1 image)
               ├── yyy.png
               ├── ...
           ├── domain2
               ├── aaa.jpg (domain2 image)
               ├── bbb.png
               ├── ...
           ├── domain3
           ├── ...
       ├── text
           ├── captions.pickle
           ├── filenames_train.pickle
           ├── filenames_test.pickle

Train

python main.py --dataset birds --phase train

Test

python main.py --dataset birds --phase test

Author

Junho Kim