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