Switch branches/tags
Nothing to show
Find file History
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
..
Failed to load latest commit information.
NoiseFill.lua
README.md
SpatialCircularPadding.lua
pollock.t7
starry.t7
stylization.lua

README.md

Realtime texture_nets

This enables realtime stylization with texture networks, for more information check https://github.com/DmitryUlyanov/texture_nets

For now 'starry night' and 'pollock' models are available, we will add more in future.

screen shot 2016-04-25 at 00 08 15

Usage

The demo depends on folding Batch Normalization into convolution in imagine-nn, to install it do:

luarocks install inn

To run on CPU do:

OMP_NUM_THREADS=2 th stylization.lua

On my dual core macbook it takes about ~0.6s to process one frame.

To run on CUDA do:

type=cuda th stylization.lua

OpenCL is supported to with type=cl. Might be slower than CPU though.

To run on input video file do:

video_path=*path to file* th stylization.lua

Model

outfile

Credits

Thanks to Dmitry Ulyanov for providing the initial version of this demo and the working network.