Artistic style transfer for mobile

It is yet another PRISMA-like model trained by us. The main objective of this project was a development of a model working with unlimited number of styles.

Problem
Development and training of model which can transfer style of random artist (e.g. Vincent Willem van Gogh) for photos from iPhone.
Solution

We used Keras framework on Tensorflow backend to develop a neural network solving this problem. The model takes as input two tensors:

  • The first tensor contains the painter's image with required style.
  • The second tensor contains the image with required content.
These two tensors are used in an iterative optimization process which generates the resulting image.