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

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