Classifying Healthy and Defective Fruits with a Multi-Input Architecture and CNN Models

Luis Chuquimarca,Boris Vintimilla,Sergio Velastin
DOI: https://doi.org/10.1109/ICPRS62101.2024.10677833
2024-10-15
Abstract:This study presents an investigation into the utilization of a Multi-Input architecture for the classification of fruits (apples and mangoes) into healthy and defective states, employing both RGB and silhouette images. The primary aim is to enhance the accuracy of CNN models. The methodology encompasses image acquisition, preprocessing of datasets, training, and evaluation of two CNN models: MobileNetV2 and VGG16. Results reveal that the inclusion of silhouette images alongside the Multi-Input architecture yields models with superior performance compared to using only RGB images for fruit classification, whether healthy or defective. Specifically, optimal results were achieved using the MobileNetV2 model, achieving 100\% accuracy. This finding suggests the efficacy of this combined methodology in improving the precise classification of healthy or defective fruits, which could have significant implications for applications related to external quality inspection of fruits.
Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
This paper aims to address the classification problem of the health status and defect status of fruits (apples and mangoes). Specifically, the study improves the accuracy of fruit classification by utilizing a multi-input architecture and Convolutional Neural Network (CNN) model, combining RGB images and contour images. The main objective of the research is to enhance the performance of the CNN model by introducing contour images, thereby more accurately distinguishing between healthy and defective fruits. This not only helps improve the efficiency of fruit quality inspection but also plays an important role in the external quality inspection of agricultural products.