taki0112/StarGAN-Tensorflow
Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)
repo name | taki0112/StarGAN-Tensorflow |
repo link | https://github.com/taki0112/StarGAN-Tensorflow |
homepage | |
language | Python |
size (curr.) | 9629 kB |
stars (curr.) | 691 |
created | 2018-06-11 |
license | MIT License |
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
Reference
Author
Junho Kim