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.
- Mixed Precision
- Feature Extraction
- Fine Tuning
- Batching & Prefetching Datasets
- ModelCheckpoint callback
- EarlyStopping callback
- ReduceLROnPlateau callback
- TensorBoard callback
The model is able to perform great achieving the accuracy of about 79.79%.