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