neural-painters-pytorch icon indicating copy to clipboard operation
neural-painters-pytorch copied to clipboard

PyTorch library for "Neural Painters: A learned differentiable constraint for generating brushstroke paintings"

The author's PyTorch implementation of Neural Painters

banner

Neural Painters: A learned differentiable constraint for generating brushstroke paintings

https://arxiv.org/abs/1904.08410

Dependencies

Dependencies are listed in environment.yaml but the notable ones are:

Notebooks

The best way to figure out how to use this code is to play around with the provided Colaboratory notebooks. We provide pre-trained neural painters.

There are runnable notebooks for the paper in the notebooks/ folder.

Since most people will probably only be interested in certain parts of the paper, we have designed them so you will be able to run each part as standalone notebooks. For example, we have provided pre-trained neural painters so you can run the style transfer notebook without having to train your own neural painter.

vae gan

  • visualizing_imagenet.ipynb - Contains code for the "Visualizing ImageNet Classes" subsection of the paper. Requires a neural painter. We provide pre-trained neural painters if you don't want to train your own.

visualize_imagenet

  • intrinsic_style_transfer.ipynb - Contains code for the "Intrinsic Style Transfer" subsection of the paper. Requires a neural painter. We provide pre-trained neural painters if you don't want to train your own.

intrinsic