All Projects → taki0112 → Stargan Tensorflow

taki0112 / Stargan Tensorflow

Licence: mit
Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)

Programming Languages

python
139335 projects - #7 most used programming language


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

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].