We public our test code for artistic style image generation to reproduce our paper results. This codebase provides: -test code -test input samples in our paper.
Please download the data and pretrain folder from: data and pretrain.
To follow anonymity requirement, download links direct to google drive under anonymous account: [email protected] .
- CUDA 10.1
- Python 3
- PyTorch tested on 1.6.0
- PIL
- skimage
- numpy
- cv2 tested on 4.2.0
We provide 5 styles in pretrain folder: cartoon, oil, wuxia, jackie, scarlett. Please use following code to generate result.
python test.py --path examples/29899.jpg --style cartoon