Awesome
CartoonGAN-Test-Pytorch-Torch
Pytorch and Torch testing code of CartoonGAN [Chen et al., CVPR18]
. With the released pretrained models by the authors, I made these simple scripts for a quick test.
Getting started
- Linux
- NVIDIA GPU
- Pytorch 0.3
- Torch
git clone https://github.com/Yijunmaverick/CartoonGAN-Test-Pytorch-Torch
cd CartoonGAN-Test-Pytorch-Torch
Pytorch
The original pretrained models are Torch nngraph
models, which cannot be loaded in Pytorch through load_lua
. So I manually copy the weights (bias) layer by layer and convert them to .pth
models.
- Download the converted models:
sh pretrained_model/download_pth.sh
- For testing:
python test.py --input_dir YourImgDir --style Hosoda --gpu 0
Torch
Working with the original models in Torch is also fine. I just convert the weights (bias) in their models from CudaTensor to FloatTensor so that cudnn
is not required for loading models.
- Download the converted models:
sh pretrained_model/download_t7.sh
- For testing:
th test.lua -input_dir YourImgDir -style Hosoda -gpu 0
Examples (Left: input, Right: output)
<p> <img src='test_img/in2.png' width=300 /> <img src='test_output/in2_Hayao.png' width=300 /> </p> <p> <img src='test_img/in3.png' width=300 /> <img src='test_output/in3_Hayao.png' width=300 /> </p> <p> <img src='test_img/5--26.jpg' width=300 /> <img src='test_output/5--26_Hosoda.jpg' width=300 /> </p> <p> <img src='test_img/7--136.jpg' width=300 /> <img src='test_output/7--136_Hayao.jpg' width=300 /> </p> <p> <img src='test_img/15--324.jpg' width=300 /> <img src='test_output/15--324_Hosoda.jpg' width=300 /> </p>Note
- The training code should be similar to the popular GAN-based image-translation frameworks and thus is not included here.
Acknowledgement
-
Many thanks to the authors for this cool work.
-
Part of the codes are borrowed from DCGAN, TextureNet, AdaIN and CycleGAN.