/style-transfer

Style transfer in TensorFlow with color preservation

Primary LanguagePython

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