Skip to content

Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)

License

Notifications You must be signed in to change notification settings

taki0112/StarGAN-Tensorflow

Repository files navigation


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

About

Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages