/Chocolate-Classification

Classify different types of Chocolates based on images from various brands such as Godiva and Lindt

Primary LanguageJupyter NotebookMIT LicenseMIT

Chocolate-Classification

Goal

An easy project to help learners understand the basics concepts of Image Classification using Convolutional Neural Network and Tensorflow. The task is to classify different types of Chocolates (Dark and White) based on images from various brands such as Godiva and Lindt. Full dataset on Kaggle: - https://www.kaggle.com/datasets/siddharthmandgi/chocolate-classification

Approach

Developed a basic Tensorflow Based CNN model

Network Architecture