Skip to content
Simple Tensorflow implementation of StarGAN (CVPR 2018 Oral)
Python
Branch: master
Clone or download
taki0112 Merge pull request #14 from MadeInChina/master
Simplify environment setup by create Dockerfile.gpu and docker-compose.yml
Latest commit 277bfc3 Jul 16, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
assests assests Jun 12, 2018
dataset/celebA dataset Jun 12, 2018
.gitignore Initial commit Jun 11, 2018
Dockerfile.gpu create Dockerfile.gpu and docker-compose.yml Apr 15, 2019
LICENSE Initial commit Jun 11, 2018
README.md Update README.md Jun 23, 2018
StarGAN.py
docker-compose.yml create Dockerfile.gpu and docker-compose.yml Apr 15, 2019
download.py
main.py clean up Jun 14, 2018
ops.py
utils.py code Jun 12, 2018

README.md


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

You can’t perform that action at this time.