Enhancing deep convolutional neural network models for orange quality classification using MobileNetV2 and data augmentation techniques

Abstract

This study introduces significant improvements in the construction of Deep Convolutional Neural Network (DCNN) models for classifying agricultural products, specifically oranges, based on their shape, size, and color. Utilizing the MobileNetV2 architecture, this research leverages its efficiency and lightweight nature, making it suitable for mobile and embedded applications. Key techniques such as Depthwise Separable Convolutions, Linear Bottlenecks, and Inverted Residuals help reduce the number of parameters and computational load while maintaining high performance in feature extraction. Additionally, the study employs comprehensive data augmentation methods, including horizontal and vertical flips, grayscale transformations, hue adjustments, brightness adjustments, and noise addition to enhance the model's robustness and generalization capabilities. The proposed model demonstrates superior performance, achieving an overall accuracy of 100% with nearly perfect precision, recall, and F1-score for both "orange_good" and "orange_bad" classes, significantly outperforming previous models which typically achieved accuracies between 70–90%. The confusion matrix shows that the model has high sensitivity and specificity, with very few misclassifications. Finally, this study empresentasizes the practical applicability of the proposed model, particularly its easy deployment on resource-constrained devices and its effectiveness in agricultural product quality control processes. These findings affirm the model in this research as a reliable and highly efficient tool for agricultural product classification, surpassing the capabilities of traditional models in this field.