Discover artwork around you using transfer learning
Find Your Art is a machine learning tool that classifies art into 25 different styles:
- Abstract Art
- Abstract Expressionism
- Art Informel
- Art Nouveau (Modern)
- Baroque
- Color Field Painting
- Cubism
- Early Renaissance
- Expressionism
- High Renaissance
- Impressionism
- Magic Realism
- Mannerism (Late Renaissance)
- Minimalism
- Naive Art (Primitivism)
- Neoclassicism
- Northern Renaissnce
- Pop Art
- Post-Impressionism
- Realism
- Rococo
- Romanticism
- Surrealism
- Symbolism
- Ukiyo-e
After scrapping / cleaning images and metadata from (https://www.wikiart.org/), a transfer learning approach was taken. Bottleneck features from the VGG16 convolutional neural network (CNN; pre-trained on the imagenet dataset) were extracted for each image, and a customized top classifier CNN was trained on the new labels.
The web app allows users to upload an image of a piece of art they like, and get the style of that image from the CNN running in the backend, as well as a recommendation of other artwork they might like. It is hosted at http://find-your-art.us/
** https://github.com/lucasdavid/wikiart
** https://github.com/chasingbob/keras-visuals