neural-style

An implementation of neural style in TensorFlow.

This implementation is a lot simpler than a lot of the other ones out there, thanks to TensorFlow's really nice API and automatic differentiation.

TensorFlow doesn't support L-BFGS (which is what the original authors used), so we use Adam. This may require a little bit more hyperparameter tuning to get nice results.

TensorFlow seems to be slower than a lot of the other deep learning frameworks out there. I'm sure this implementation could be improved, but it would probably take improvements in TensorFlow itself as well to get it to operate at the same speed as other implementations. As of now, it seems to be around 3x slower than implementations using Torch.

Running

python neural_style.py --content <content file> --styles <style file> --output <output file>

(run python neural_style.py --help to see a list of all options)

Example 1

Running it for 500-2000 iterations seems to produce nice results. With certain images or output sizes, you might need some hyperparameter tuning (especially --content-weight, --style-weight, and --learning-rate).

The following example was run for 1000 iterations to produce the result (with default parameters):

These were the input images used (me sleeping at a hackathon and Starry Night):

Example 2

The following example demonstrates style blending, and was run for 1000 iterations to produce the result (with style blend weight parameters 0.8 and 0.2):

The content input image was a picture of the Stata Center at MIT:

The style input images were Picasso's "Dora Maar" and Starry Night, with the Picasso image having a style blend weight of 0.8 and Starry Night having a style blend weight of 0.2:

Running in Docker(CPU only)

Docker build:

docker build -t docker_neural_style .

Download Pre-trained imagenet-vgg-verydeep-19.mat in your computer

wget http://www.vlfeat.org/matconvnet/models/beta16/imagenet-vgg-verydeep-19.mat

Store in /Users/you/ for example

Docker run

docker run -it -p 8888:8888 -v /Users/you:/notebooks/neural-style-mat docker_neural_style

Then you can run neural-style code with jupyter notebook in http://localhost:8888/

Docker run in command

docker run -it -p 8888:8888 -v /Users/you:/notebooks/neural-style-mat docker_neural_style /bin/bash

python neural_style.py --content examples/1-content.jpg --styles examples/1-style.jpg --output examples/myoutput.jpg --network ../neura
l-style-mat/imagenet-vgg-verydeep-19.mat

Requirements

TensorFlow
NumPy
SciPy
Pillow
Pre-trained VGG network (MD5 8ee3263992981a1d26e73b3ca028a123) - put it in the top level of this repository

License

Copyright (c) 2015-2016 Anish Athalye. Released under GPLv3. See LICENSE.txt for details.

License

lijingpeng/neural-style

About

Resources

License

Uh oh!

Stars

Watchers

Forks

Releases

Packages 0

Languages

Packages