murilobd / paint-like-a-pro

Paint like a pro implements the pix2pix machine learning model

Paint like a pro

This webapp implements the pix2pix model generated using tensorflow (and it's highly inspired from affinelayer implementaion).

The step to resize images and get only the edges of it was done by using similar techniques fromyining1023.

The style transfer (to enable it, just pass transferStyle=true to URL as query param) was copy/paste from https://github.com/reiinakano/arbitrary-image-stylization-tfjs. Thanks for that awesome model that just works.

All images for the model were downloaded from https://en.most-famous-paintings.com/MostFamousPaintings.nsf/ListOfTop1000MostPopularPainting.

Thanks to Christopher Hesse for this blog post showing a pix2pix in action! Was super inspiring!

All the other stuffs (html?) were done by me! Tks!

How to use it

Draw in the 1st canvas and press the 'paint' button right next to it. See the magic! If you want to load an image, you can too. But mind the image should be a sketch (it means white background and black lines).

You can test it from this https://paint-like-a-pro.netlify.app/

If you want to test in your computer, is simple as forking this repo and serving the index.html file.