Simple chocolate classifier that is able to classify given image of a chocolate between 12 different available categories:
- Coffee Crisp
- Ferrero Rocher
- Kit Kat
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.