Mango production is fundamental to the agricultural economy, generating income and employment in various communities. Accurate estimation of its production optimizes the planning and logistics of harvesting; traditionally, manual methods are inefficient and prone to errors. Currently, machine learning, by handling large volumes of data, emerges as an innovative solution to enhance the precision of mango production estimation. This study presents an analysis of mango fruit detection using machine learning algorithms, specifically YOLO version 8 and Faster R-CNN. The present study employs a dataset consisting of 212 original images, annotated with a total of 9604 labels, which has been expanded to include 2449 additional images and 116,654 annotations. This significant increase in dataset size notably enhances the robustness and generalization capacity of the model. The YOLO-trained model achieves an accuracy of 96.72%, a recall of 77.4%, and an F1 Score of 86%, compared to the results of Faster R-CNN, which are 98.57%, 63.80%, and 77.46%, respectively. YOLO demonstrates greater efficiency, being faster in training, consuming less memory, and utilizing fewer CPU resources. Furthermore, this study has developed a web application with a user interface that facilitates the uploading of images from mango trees considered samples. The YOLO-trained model detects the fruits of each tree in the representative sample and uses extrapolation techniques to estimate the total number of fruits across the entire population of mango trees.
Read full abstract