Sign inSign up

oneoffcoder/dl-transfer

By oneoffcoder

Updated over 6 years ago

An executable Docker image for use with neural style transfer using PyTorch.

Image
0

496

oneoffcoder/dl-transfer repository overview

This project is an executable container for Neural Style Transfer. At a high level, you supply content and style images as inputs, and the application will mix the style and content to produce a new image that is as faithful as possible to the style in the style image and the content in the content image.

GitHub

Code

Usage

To use the container, type in the following. Note that the -s flag specifies the style image and -c specifies the content image. By default, an image/output.jpg file be created. To control the output image, specify -o (e.g. -o image/final.jpg).

docker run -it \
    -v `pwd`/image:/app/image \
    --gpus=all \
    --shm-size=5g \
    oneoffcoder/dl-transfer -s image/picasso-01.jpg -c image/dancing.jpg

Citation

@misc{oneoffcoder_dl_transfer_2019, 
title={An executable docker container with Neural Style Transfer}, 
url={https://github.com/oneoffcoder/docker-containers/tree/master/dl-transfer}, 
journal={GitHub},
author={One-Off Coder}, 
year={2020}, 
month={Jan}}

Tag summary

Content type

Image

Digest

Size

5.9 GB

Last updated

over 6 years ago

docker pull oneoffcoder/dl-transfer