Multi-Class Image Classification Model for Mango Varieties Classification Using TensorFlow Take 1

Template Credit: Adapted from a template made available by Dr. Jason Brownlee of Machine Learning Mastery.

SUMMARY: This project aims to construct a predictive model using a TensorFlow convolutional neural network (CNN) and document the end-to-end steps using a template. The Mango Varieties Classification dataset is a multi-class classification situation where we attempt to predict one of several (more than two) possible outcomes.

INTRODUCTION: This dataset contains 1,600 images of eight varieties of Pakistani mangoes. The research team performed experiments on the dataset to create an automated classification and grading system. The system classifies the harvested mangoes for farmers to deliver high-quality mangoes for export with high accuracy.

ANALYSIS: The ResNet50V2 model’s performance achieved an accuracy score of 99.10% after 20 epochs using the training dataset. When we applied the model to the validation dataset, the model achieved an accuracy score of 88.13%.

CONCLUSION: In this iteration, the TensorFlow ResNet50V2 CNN model appeared suitable for modeling this dataset.

Dataset ML Model: Multi-Class classification with numerical features

Dataset Used: Mango Varieties Classification and Grading, Rizwan Iqbal, Hafiz Muhammad; Hakim, Ayesha (2021), “Mango Variety and Grading Dataset,” Mendeley Data, V1, DOI: 10.17632/5mc3s86982.1

Dataset Reference:

One source of potential performance benchmarks:

The HTML formatted report can be found here on GitHub.