weishancc/fast_neural_style repository overview

⁠fast-neural-style (cuda)

This is an image based on https://github.com/pytorch/examples/tree/master/fast_neural_style⁠.
When container is up, model is served on flask⁠ server (0.0.0.0:5000)

[Note]

• It eliminated training process and only for stylizing to mosaic style.
• It's cuda10 version, have your gpu prepare and install nvidia-docker⁠ first : )

⁠How to use?

$ docker pull weishancc/fast_neural_style:latest
$ docker run -it \
--runtime=nvidia \
--name fst \
--network=host \
weishancc/fast_neural_style:latest

⁠Test with Postman

POST request
http://YOUR_IP:5000/mosaic 
application/json {"image":  IMAGE_URL}

⁠References

• fns⁠
• docker-pytorch⁠