Automatic Classification of the Ripeness Stage of Mango Fruit Using a Machine Learning Approach

Abstract

Most mango farms classify the maturity stage manually by trained workers using external indicators such as size, shape, and skin color, which can lead to human error or inconsistencies. We developed four common machine learning (ML) classifiers, the k-mean, naïve Bayes, support vector machine, and feed-forward artificial neural network (FANN), all of which were aimed at classifying the ripeness stage of mangoes at harvest. The ML classifiers were trained on biochemical data and then tested on physical and electrical data.The performance of the ML models was compared using fourfold cross validation. The FANN classifier performed the best, with a mean accuracy of 89.6% for unripe, ripe, and overripe classes, when compared to the other classifiers.