/StarGAN-Tensorflow

Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)

Primary LanguagePythonMIT LicenseMIT

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

Related works

Reference

Author

Junho Kim