/spatial_style_transfer

This is package for doing spatially controlled style transfer in PyTorch.

Primary LanguageJupyter Notebook

Spatial Style Transfer

This is package for doing spatially controlled style transfer in PyTorch. This project is based off on the style transfer tutorial of PyTorch and a paper about spatial control in style transfer.

One can easy utilize the package as demonstrated in the notebook.

One example, when one applies a mask for the left side and upper right corner.

output_1

Another example, when one applies a mask for a circle in the middle of the upper half and the lower left corner.

output_2