jimmyyhwu / style-transfer

Style transfer in TensorFlow with color preservation

style-transfer

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

Style Transfer with Color Preservation

Usage

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>

Requirements

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