Public Repository

Last pushed: a year ago
Short Description
Dockerized neural-style algorithm
Full Description

Neural Style

A dockerized version of the neural style algorithm by jcjohnson. nvidia-docker is used to make use of GPU hardware.

How to use this image

You can directly invoke the core neural-style algorithm by simply running a container of this image, for example:

nvidia-docker run --rm albarji/neural-style -h

produces the usage help.

To apply the neural-style method on some host images, map the host folder with such images to the container /images folder through a volume such as

nvidia-docker run --rm -v $(pwd):/images albarji/neural-style -backend cudnn -cudnn_autotune -content_image content.png -style_image style.png

The container uses as work directory the /images folder, so the results will be readily available at the mounted host folder.

In order to take full advantage of the cudnn libraries (also included in the image) the options -backend cudnn -cudnn_autotune are always recommended.

As an example, let's redraw Docker's logo in the famous style of Van Gogh's Starry Night:

nvidia-docker run --rm -v $(pwd):/images albarji/neural-style -backend cudnn -cudnn_autotune -content_image contents/docker.png -style_image styles/vangogh.jpg

Helper scripts

Check the project on github for some helper scripts that simplify the creative process.

Docker Pull Command

Comments (0)