Style-Transfer-with-CNN

Simple wrapper to make style transfer with pre-trained CNN VGG16

Most of code taken from here

Example of usage:

from style_transfer import VGG16StyleTransfer

model = VGG16StyleTransfer()
model.load_pictures('base.jpg', 'style.jpg')
model.transfer_style(iterations=20, output_file='result_picture')

Medium article:

https://medium.com/cindicator/style-transfer-with-neural-network-weekend-of-a-data-scientist-8009d2285b74