A Single Stream Modified MobileNet V2 and Whale Controlled Entropy Based Optimization Framework for Citrus Fruit Diseases Recognition

Abstract

Fruit disease recognition is quickly becoming a hot topic in the field of computer vision. The presence of plant diseases not only reduces fruit production but also causes a significant loss to the national economy. Citrus fruits help to strengthen the immune system, allowing it to fight off diseases such as COVID-19. Manual inspection of fruit diseases with the naked eye takes time and is difficult; therefore, a computer based method is always required for accurate recognition of plant diseases. Several deep learning techniques for recognizing citrus fruit diseases have been introduced in the literature. Existing techniques had several issues, including redundant features, convolutional neural network (CNN) model selection, low contrast images, and long computational times. In this paper, a single stream convolutional neural network architecture is proposed for recognizing citrus fruit diseases. In the first step, data augmentation is performed using four contrast enhancement operations: shadow removal, adjusting pixel intensity, improving brightness, and improving local contrast. The MobileNet-V2 CNN model is selected and fine-tuned in the second step. Using the transfer learning process, the fine-tuned model is trained on the augmented citrus dataset. The newly trained model is used for deep feature extraction; however, analysis shows that the extracted deep features contain little redundant information. As a result, an improved Whale Optimization Algorithm (IWOA) is used in the third step. The best features are then classified using machine learning classifiers in the final step. The augmented citrus fruits, leaves, and hybrid dataset were used in the experimental process and achieved an accuracy of 99.4, 99.5, and 99.7%. When compared to existing techniques, the proposed architecture outperformed them in terms of accuracy and time.