April 30, 2019

125 words 1 min read

taki0112/StarGAN-Tensorflow

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

Summary

overview

Results (128x128, wgan-gp)

Women

women

Men

men

Reference

Author

Junho Kim

comments powered by Disqus