Skip to content

mhd-danish/porfolio_food_vision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Food Vision Project

Food Vision Project is multi-classification based project which is able to classify 101 classes of food101 dataset.

Approach

  1. Importing food101 dataset.
  2. Preprocessing the data.
  3. Using TensorFlow to prepare the dataset.
  4. Exploring the data.
  5. Creating model callbacks.
  6. Importing EfficientNet-B2 architecture for transfer learning.
  7. Fine-Tuning the model.
  8. Experimenting with Data Augmentation.
  9. Evaluating the model (Finding our model's most wrong predictions).
  10. Making predictions with our Food Vision model on custom images of food.

Why Choose EfficientNet-B2?

Pre-Trained CNN Models

To get better results and to save time, transfer learning is used to train the model for this project. EfficientNet-B2 that was trained on ImageNet Dataset was used as a feature extractor for the model.

EfficientNet-V2M could've also been considered but it wasn't chosen for several reasons. The Google Colaboratory's GPU Tesla T4 was used for this project. Although, this wasn't a very low-end GPU but when training a model with larger parameters, having just one won't suffice. So, the following requirements were taken under consideration when choosing the architecture:

  1. Lower Depth (Faster Gradient Descent).
  2. Higher the ratio accuracy / no. of parameters, the better (Faster Training Epochs).
  3. The model's predecessors have proven to show good results on the considered dataset.

Best Model

The FineTuned EfficientNet-B2 Model has the best acccuracy among the three models.

Models' Performance

Results

The following are couple of the results (correct ones) generated by the model.

alt text alt text alt text alt text alt text alt text

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published