Official Tensorflow implementation Open ! -> Clova AI
StarGAN v2 β Un-official TensorFlow Implementation [Paper] [Pytorch]
: Diverse Image Synthesis for Multiple Domains
Overview
Architecture
Generator |
Mapping Network |
Style Encoder & Discriminator |
|
|
|
Usage
βββ dataset
Β Β βββ YOUR_DATASET_NAME
Β Β βββ train
Β Β Β Β Β βββ domain1 (domain folder)
βββ xxx.jpg (domain1 image)
βββ yyy.png
βββ ...
βββ domain2
βββ aaa.jpg (domain2 image)
βββ bbb.png
βββ ...
βββ domain3
βββ ...
Β Β βββ test
βββ zzz.jpg (any content image)
βββ www.png
βββ ...
Train
python main.py --dataset celebA-HQ_gender --phase train --batch_size 2 --gpu_num 4
Test
python main.py --dataset celebA-HQ_gender --phase test
Refer test
python main.py --dataset celebA-HQ_gender --phase refer_test --refer_img_path refer_img.jpg
Our results
Author
Junho Kim