PyTorch-OptimalStyleTransfer

This is an unofficial PyTorch implementation of paper A Closed-form Solution to Universal Style Transfer - ICCV 2019

1. Colab demo
2. Official implementation (Lua)

Some examples:

Comparison with other methods:

Installation

1. Install packages pip install -r requirements.txt

2. Download model weights and extract archive into models/ folder

Style Transfer

You can test model in demo.ipynb or use a script:

python test.py --content=content/brad_pitt.jpg --style=style/picasso_self_portrait.jpg