Skip to content

Using machine learning to classify art/doodles as my (Nitisha's) art style or not.

Notifications You must be signed in to change notification settings

nitisha121/doodle_ai

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

33 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Doodle AI NST

Contributors: Amirmehdi Sharifzad & Nitisha Agarwal Resources: Coursera CNN course, Gatsy's NST Algorithm, TensorFlow style transfer

Neural style transfer is an optimization technique used to take two images—a content image C and a style reference image S and blend them together so the output image looks like the content image, but “painted” in the style of the style reference image.

This is implemented by optimizing the output image to match the content statistics of the content image and the style statistics of the style reference image. These statistics are extracted from the images using a convolutional network.

An example of its use is shown below:

We used a pre-trained ConvNet model with 19 layers - VGG-19 - which is stored as a Python dictionary -- key:variable name, value: tensor with variable value.

The purpose of this project was to transpose my doodling style onto simple images, such as flowers. This is one of the earlier examples:

C: An image of a flower S: Doodle of an elephant G (generated): Flower image - more greyscale and some transposed details.

How does Style Transfer Work?

The NST algorithm:

  • Images are inputted into CNN
  • Intermediate layers of the model is used to get the content and style representations of the image.
  • The Style image cost and the Content image cost are calculated to sum to total cost
  • The total cost of each generated image G is calculated and minimized at each iteration
  • The generated image G is optimized until max number of iterations specified is reached

About

Using machine learning to classify art/doodles as my (Nitisha's) art style or not.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published