Banana quality classification using lightweight CNN model with microservice integration system

Abstract

Banana sorting has been performed manually, which often leads to human error due to the high volume and diverse characteristics involved. This paper presents a banana quality classification system using ConsolutechMobileNetV2 (CST-MobileNetV2) to classify banana ripeness into four categories unripe, ripe, overripe, and rotten. A lightweight deep learning model is proposed and integrated with a uniquely designed microservice system to optimize performance while minimizing computational demands. A publicly available dataset containing 13,478 images was used, and the data split into 56% for training, 14% for validation, and 30% for testing. Image normalization and augmentation techniques were applied to enhance the model's robustness. The model's performance was evaluated using a confusion matrix, achieving 98% precision, recall, and F1-score. The proposed model was compared with other deep learning models to benchmark its performance and deployed in different operating systems to evaluate its flexibility and capabilities. The LINE platform was employed as the user interface, enabling practical interaction with users. The system also demonstrated an average response time of 9.25 seconds per image, ensuring efficient processing, delivers high accuracy and scalability making it a practical and efficient solution for automated banana quality classification.