https://github.com/fzliu/style-transfer

An implementation of "A Neural Algorithm of Artistic Style" by L. Gatys, A. Ecker, and M. Bethge. http://arxiv.org/abs/1508.06576.
https://github.com/fzliu/style-transfer

Last synced: 13 days ago
JSON representation

An implementation of "A Neural Algorithm of Artistic Style" by L. Gatys, A. Ecker, and M. Bethge. http://arxiv.org/abs/1508.06576.

• Host: GitHub
• URL: https://github.com/fzliu/style-transfer
• Owner: fzliu
• Created: 2015-09-09T08:59:20.000Z (about 9 years ago)
• Default Branch: master
• Last Pushed: 2017-08-30T05:57:48.000Z (about 7 years ago)
• Last Synced: 2024-10-16T13:05:11.042Z (26 days ago)
• Language: Python
• Size: 1.67 MB
• Stars: 1,537
• Watchers: 104
• Forks: 387
• Open Issues: 29
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# style-transfer

## Introduction

This repository contains a pyCaffe-based implementation of "A Neural Algorithm of Artistic Style" by L. Gatys, A. Ecker, and M. Bethge, which presents a method for transferring the artistic style of one input image onto another. You can read the paper here: http://arxiv.org/abs/1508.06576. 

Neural net operations are handled by Caffe, while loss minimization and other miscellaneous matrix operations are performed using numpy and scipy. L-BFGS is used for minimization.

## Requirements

 - Python >= 2.7

 - CUDA >= 6.5 (highly recommended)

 - Caffe

CUDA will enable GPU-based computation in Caffe.

## Download

To run the code, you must have Caffe installed and the appropriate Python bindings in your `PYTHONPATH` environment variable. Detailed installation instructions for Caffe can be found [here](http://caffe.berkeleyvision.org/installation.html).

All of the necessary code is contained in the file `style.py`. You can try it on your own style and content image by running the following command:

```

python style.py -s  -c  -m  -g 0

```

The prototxts which come with the vanilla Caffe install aren't quite compatible with this code - working ones have already been added to this repository as a result of this. To get the pretrained models, simply run:

```

bash scripts/download_models.sh

```

This will grab the convnet models from the links provided in the [Caffe Model Zoo](https://github.com/BVLC/caffe/wiki/Model-Zoo). You may also specify the exact model you'd like to download by running:

```

bash scripts/download_models.sh 

```

Here, `` must be one of `vgg16`, `vgg19`, `googlenet`, or `caffenet`.

## Sample

Original images: [San Francisco](https://www.flickr.com/photos/anhgemus-photography/15377047497) by Anh Dinh, [Nanjing in winter snow, 2008](https://www.flickr.com/photos/emmajg/3199018106) by Emma Gawen, and [Blade Runner's Johannesburg](https://www.flickr.com/photos/andryn2006/19114016482) by Andrew Moore. All images were released under the Creative Comments license. Each output image was initialized with the content image, and 500 BFGS iterations under the VGG model were performed in each instance.































These results can also be found in the `images` folder in the repository root.

A more in-depth set of examples can be found [here](http://frankzliu.com/artistic-style-transfer/).