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

504

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