Skip to content

alex-ele-xela/Neural-Style-Transfer

Repository files navigation

Neural Style Transfer

Language Framework

About

This project aims to explore the domain of Neural Style Transfer by recreating the work of Gatys et al., 2015

WIP

Creating a neural network to convert the style of an image to a pre-defined style based on the work of Johnson et al., 2016

Collaborators

Alex Prateek Shankar

Sneh Shah

What is Neural Style Transfer

A method used in artificial intelligence and computer vision called neural style transfer (NST) enables the creative style of one image—referred to as the "style image"—to be transferred to another—referred to as the "content image." The outcome is a brand-new image that blends the style and content of the content image.

Neural style transfer involves keeping the high-level content of the content image while utilizing a neural network to repeatedly change the pixels of the content image to fit the style of the style image. The network is typically pretrained on a large dataset for image recognition tasks, such as the VGG network.

Examples

The following images are generated using the Style Transfer optimization, with Content Weight set to 1e5 and Style Weight set to 1e4. All are generated using LBFGS optimizer and correspond to the 500th iteration of the optimization process

Content Image Style Image Generated Image
tubingen Starry Night Output
tubingen Weeping Woman Output
tubingen Mosaic Output
Mona Lisa Starry Night Output
Mona Lisa Weeping Woman Output
Mona Lisa Mosaic Output
Lion Mosaic Output

How to access all examples

  • The content images are available at data/content_images
  • The style images are available at data/style_images
  • All the generated examples along with logs and video are available at data/output_images/optimization/<content image> styled as <style image>/<content weight> <style weight> <tv weight>/

For example:

Tubingen styled as Van Gogh Starry Night