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