Skip to content
main
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

README.md

chocolate-classifier Binder

Simple chocolate classifier that is able to classify given image of a chocolate between 12 different available categories:

  • Aero
  • Coffee Crisp
  • Ferrero Rocher
  • Ghirardelli
  • Hershey's
  • Kinder
  • Kit Kat
  • Lindor
  • Reese's
  • Snickers
  • Toblerone
  • Twix

Model export.pkl has been created via fine tuning on a pretrained model on ImageNet on around 100-300 images of each chocolate kinds. The model has shown around 95% accuracy on various validation sets.

About

Simple chocolate classifier that is able to classify given image of a chocolate between 12 different available categories

Topics

Resources