CartoonGAN-Tensorflow
Simple Tensorflow implementation of CartoonGAN (CVPR 2018)
Pytorch version
Requirements
- Tensorflow 1.8
- Python 3.6
Usage
1. Download vgg19
2. Do edge_smooth
> python edge_smooth.py --dataset face2anime --img_size 256
โโโ dataset
ย ย โโโ YOUR_DATASET_NAME
ย ย โโโ trainA
ย ย ย ย ย โโโ xxx.jpg (name, format doesn't matter)
โโโ yyy.png
โโโ ...
ย ย โโโ trainB
โโโ zzz.jpg
โโโ www.png
โโโ ...
ย ย โโโ trainB_smooth (After you run the above code, it will be created automatically)
ย ย โโโ zzz.jpg
โโโ www.png
โโโ ...
ย ย โโโ testA
โโโ aaa.jpg
โโโ bbb.png
โโโ ...
3. Train
- python main.py --phase train --dataset face2anime --epoch 100 --init_epoch 1
4. Test
- python main.py --phase test --dataset face2anime
Author
Junho Kim