/caffe-neural-style

Caffe implementation of the paper "A Neural Algorithm of Artistic Style" by Leon A. Gatys, Alexander S. Ecker, and Matthias Bethge.

Primary LanguagePython

Neural Style

Setup

  1. Build a modified version of Caffe, which provides:

    • Gram layer
    • Total Variation Loss layer
    • L-BFGS solver

  2. Install python dependencies:

    pip install -r requirements.txt

  3. Get the VGG model:

    cd vgg; ./get_model.sh