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