Skip to content

Attempts to guess an image's artistic style using a deep neural network

Notifications You must be signed in to change notification settings

noahvrz/artistic_style

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Artistic Style Recognition

Attempts to guess a painting's artistic style using a deep neural network. Uses transfer learning on an optimized convolutional SqueezeNet applied to data (not included) from the WikiArt database.

Built for a Machine Learning course at Oxford University.

Accuracy vs. Epoch Loss vs. Epoch

TODO:

  • Add ability to use multiple architectures, not just SqueezeNet
  • Add CUDNN benchmarking
  • Add ability to properly disable CUDA (try import?)

About

Attempts to guess an image's artistic style using a deep neural network

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages