taki0112 / StarGAN-Tensorflow

Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)

---

Requirements

• Tensorflow 1.8
• Python 3.6

Usage

Downloading the dataset

> python download.py celebA

├── dataset
   └── celebA
       ├── train
           ├── 000001.jpg 
           ├── 000002.jpg
           └── ...
       ├── test (It is not celebA)
           ├── a.jpg (The test image that you wanted)
           ├── b.png
           └── ...
       ├── list_attr_celeba.txt (For attribute information) 

Train

• python main.py --phase train

Test

• python main.py --phase test
• The celebA test image and the image you wanted run simultaneously

Pretrained model

• Download checkpoint for 128x128

Summary

Results (128x128, wgan-gp)

Women

Men

Related works

• CycleGAN-Tensorflow
• DiscoGAN-Tensorflow
• UNIT-Tensorflow
• MUNIT-Tensorflow

Reference

• StarGAN paper
• Author pytorch code

Author

Junho Kim