This is a TensorFlow implementation of style transfer as described in A Neural Algorithm of Artistic Style. This also implements a variation of the color preservation strategy described in Preserving Color in Neural Artistic Style Transfer.

Basic style transfer demo:

python main.py

Style transfer with color preservation demo:

python main.py --preserve_color

Style transfer on custom images:

python main.py --content <custom_content.jpg> --style <custom_style.jpg> --output <output_dir>

• tensorflow 0.12
• cv2
• scikit-image
• scipy
• pretrained VGG-19 npy model from tensorflow-vgg