/PyTorch-OptimalStyleTransfer

An unofficial PyTorch implementation of paper "A Closed-form Solution to Universal Style Transfer - ICCV 2019"

Primary LanguageJupyter Notebook

PyTorch-OptimalStyleTransfer

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

  1. Colab demo google colab logo
  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