Awesome

Install

conda install -c pytorch pytorch torchvision
conda install cython scipy

Download pre-trained model here

Put it under ./checkpoint_model

Please change the images you want to use in the following folders:

inputs/content: a folder to store all content images
inputs/style: a folder to store all style images

Then run

python demo_sttr_image.py

You could see results in outputs/test_outputs/0005