Disease detection and classification in pomegranate fruit using hybrid convolutional neural network with honey badger optimization algorithm

Abstract

Fruit cultivation plays a pivotal role in improvement of the agricultural economy. Pomegranate is a nutritionally rich fruit that is highly valuable because of its excellent antioxidant properties, richness in vitamins and fiber. Pomegranate is affected by diseases, which can severely affect its yield and quality. Timely disease detection is vital for effective disease control in pomegranate orchards. This study presents a novel approach for disease detection in pomegranate and categorization of pomegranate using a hybrid Convolutional Neural Network (CNN) coupled with the Honey Badger Optimization Algorithm (HBOA). The proposed hybrid CNN-HBOA approach leverages the strengths of both CNN and metaheuristic optimization algorithm HBOA. The CNN model was designed to extract high-level features from images of pomegranate affected by various diseases, and HBOA was employed to fine-tune the CNN parameters to improve the accuracy and robustness of the model. Initially, the quality of the images was enhanced using a contrast enhancement technique, and the images were segmented using the Detectron2 algorithm. Finally, the extracted features were fed as input to the disease detection stage wherein the CNN classifies the disease. Furthermore, the classification accuracy of the CNN model was enhanced using the HBOA algorithm. The performance of the hybrid CNN-HBOA algorithm was evaluated using a vast collection of images depicting multiple disease categories of pomegranate. Experimental results demonstrate the superiority of the hybrid CNN-HBOA system, which achieves a detection accuracy of 99.58%, precision of 100%, recall of 99.71% and F1 score of 99.75% compared to other existing state-of-the-art models.

Keywords

• Convolutional neural network
• honey badger optimization algorithm
• pomegranate fruit
• confusion matrix
• disease classification