Skip to content

This repository contains work on Food Dishes Classification from Images, using a truncated version of Food-101 dataset, having 20 categories, using Keras API of TensorFlow in Python.

Notifications You must be signed in to change notification settings

neelabhsinha/Food-Dishes-Classification-using-Deep-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 

Repository files navigation

Food-Dishes-Classification

This repository contains work on Food Dishes Classification from Images, using a truncated version of Food-101 dataset, having only 20 categories, using Keras API of tensorflow in Python.

Dataset - Food 101 This data was truncated to use 20 categories for the classification. The number of images used in training and testing set, and the labels used can be seen from the following graph -

Model -

This model is built in such a way that initial layers are that of the VGG-16 Model, pre-trained on imagenet weights. The initial layers are frozen for training, allowing us to fine-tune the later layers for effective feature recognition. The recognized features are then fed to a dense network for classification.

Training -

The results of training accuracies vs. epochs are shown below. Complete details of architecture and results can be viewed from the Visualizations folder.

Results & Conclusions -

S. No. Metric Value
1 Accuracy 52.2%
2 Top 5 Accuracy 82.03%
  • VGG16 model identifies features effectively because of more depth in the architecture, which allows it to identifiy both lower level and higher level features
  • Transfer learning helps us to make the training faster as the initial layers are used to identify local features only, which are same for almost all models. So, freezing the initial layers allows us to speed-up the training remarkably
  • Features like dropout, regularization, etc. allows us to make the model more robust

Steps to execute the code in this reposirory:

  1. Install Anaconda Distribution and all relevant libraries

  2. Prepare the dataset

  3. Open up the terminal and type -

$git clone https://github.com/neelabhsinha/Food-Dishes-Classification-using-Deep-Learning.git
$cd Food-Dishes-Classification-using-Deep-Learning
$jupyter notebook

About

This repository contains work on Food Dishes Classification from Images, using a truncated version of Food-101 dataset, having 20 categories, using Keras API of TensorFlow in Python.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published