harshd23 / Food_Vision

The purpose of Food Vision project is to classify 101 variety of food items using Machine Learning.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

FOOD VISION:


Introduction:

Food Vision is a simple project which uses Machine Learning algorithms to train the computer to classify 101 classes of food items. The dataset is available from the TensorFlow datasets library. Also, the dataset can be found on Kaggle.The goal is to be able to automatically classify an unknown image using the dataset. The model created in this project was using transfer learning algorithm. The model used for Food Vision project was Efficient Net B0.

Features of Food Vision:

  • Mixed Precision
  • Feature Extraction
  • Fine Tuning
  • Batching & Prefetching Datasets
  • ModelCheckpoint callback
  • EarlyStopping callback
  • ReduceLROnPlateau callback
  • TensorBoard callback

Model Training curves using TensorBoard:

Different Models Epoch Accuracy Epoch Loss

Resources:

Conclusion:

The model is able to perform great achieving the accuracy of about 79.79%.

About

The purpose of Food Vision project is to classify 101 variety of food items using Machine Learning.

License:MIT License


Languages

Language:Jupyter Notebook 52.0%Language:PureBasic 47.5%Language:Python 0.6%