Spenhouet / automated-deep-photo-style-transfer

TensorFlow implementation for the paper "Automated Deep Photo Style Transfer"
https://arxiv.org/abs/1901.03915
111 stars 36 forks source link
deep-learning neural-networks segmentation style-transfer tensorflow

Automated Deep Photo Style Transfer

This repository holds a TensorFlow implementation for the paper Automated Deep Photo Style Transfer.

At its core this is a TensorFlow based implementation of the paper Deep Photo Style Transfer.

One of the main contributions of “Automated Deep Photo Style Transfer” is the automatic segmentation of input images and a semantic grouping thereof. Another contribution of this is the optimization of the transfer image by improving the aesthetics of the image with the use of Neural Image Assessment (NIMA).

Examples

Given a content and style image, automatically a segmentation is created and semantically grouped to produce a transfer image in the size of the content image by using the Deep Photo Style Transfer:

Overview

Here are some example results (from left to right are the content image, the resulting transfer image and the style image):

Example results

Setup

  1. Download or clone repository files to your computer
  2. Go into repository folder
  3. Install requirements: pip3 install -r requirements.txt --upgrade
  4. Download the weights.zip from the latest release and unzip it into a new folder weights under the project root.

Acknowledgement

Disclaimer

This software is published for academic and non-commercial use only.