CartoonGAN

Torch implementation for Yang Chen, Yu-Kun Lai, Yong-Jin Liu. CartoonGAN: Generative Adversarial Networks for Photo Cartoonization, CVPR2018

This code borrows from early version of CycleGAN.

Installation

The same as CycleGAN

Install torch and dependencies from https://github.com/torch/distro
Install torch packages nngraph, class, display

luarocks install nngraph
luarocks install class
luarocks install https://raw.githubusercontent.com/szym/display/master/display-scm-0.rockspec

Clone this repo:

git clone https://github.com/FlyingGoblin/CartoonGAN.git
cd CartoonGAN

Apply pre-trained Model

Download the pre-trained model:

bash ./pretrained_models/download_model.sh

Generate Shinkai style images:

DATA_ROOT=./datasets/<your test data> name=Shinkai model=one_direction_test phase=test loadSize=256 fineSize=256 resize_or_crop="scale_width" th test.lua

Dataset

For the photo: All the images in our paper are from CycleGAN: https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix (XXX2photo).
For the Cartoon image: Because all the cartoon images are from films, I don't know if I have the right to upload them. I don't know if there will be any copyright problem, I am sorry.
- I just calculate the SSIM and PSNR between frames to remove too samiliar images.
- The cartoon dataset should not be too small.

Train

Dataset: Prepare your datasets in ./datasets/, it should have subfolders train_A, train_B, train_B_edge
Train a model with init process:

DATA_ROOT=./datasets/<your data> name=<your name> th train.lua

Train a model with an init model:

DATA_ROOT=./datasets/<your data> name=<your name> init_model=<path to your init model> th train.lua

Test

Dataset: Prepare your datasets in ./datasets/, it should have subfolders test_A

DATA_ROOT=./datasets/<your test data> name=<your name> model=one_direction_test phase=test loadSize=256 fineSize=256 resize_or_crop="scale_width" th test.lua

BlaBla

My email: chenyang15@mails.tsinghua.edu.cn can NOT send emails because of my graduation, and it won't be able to receive email very soon (maybe I have already missed some emails). So please contact me by cylily93@gmail.com.
In fact I didn't mean to clean up the code before because of laziness _(:з」∠)_, but after receiving several contact asking me about the reproduction, I decided to clean them up to avoid "rebuilding the wheel".
I hope this code can speed up your work ~
发现我这个作者的代码的星星在GitHub里CartoonGAN相关项目里才排第5啊_(:з」∠)_，而且和前面的差好远_(:з」∠)_我唯一一个有可能有星星的项目啊_(:з」∠)_哭o(╥﹏╥)o