jxcodetw / Stylebank

Unofficial Implementation of the paper: StyleBank: An Explicit Representation for Neural Image Style Transfer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Stylebank

Implementation of the paper: StyleBank: An Explicit Representation for Neural Image Style Transfer

Implemented

  • Training
  • Incremental Learning

Training

  1. Prepare your content image dataset (I use MS-COCO)
  2. Prepare your style image dataset

It takes about 1 days to train on a GeForce GTX 1080 Ti. Pre-trained weights are provided. Just load the encoder, decoder and the stylebank you want.

Examples

Style Source Transfered

About

Unofficial Implementation of the paper: StyleBank: An Explicit Representation for Neural Image Style Transfer


Languages

Language:Jupyter Notebook 99.9%Language:Python 0.1%