Skip to content

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

License

Notifications You must be signed in to change notification settings

harshd23/Food_Vision

Repository files navigation

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%.