A Multi-kernel CNN model with attention mechanism for classification of citrus plants diseases.

Abstract

One of the primary challenges leading to a significant reduction in agricultural production is the prevalence of diseases affecting citrus plants. Prevention and monitoring the spread of citrus plant diseases is crucial for maintaining citrus production. This decrease in productivity adversely affects the overall economy. The essential step for enhancing the quality of fruit production and promoting economic growth involves the classification and identification of leaf diseases in the early stage. In this work, a multi-kernel CNN model with attention mechanism is used for classification of citrus plants diseases is proposed. Initially, the input image is pre-processed for resizing the images as the images are obtained from different datasets. After resizing the image, the feature extraction process is carried out by the pretrained convolutional neural networks. In the next step, the two attention mechanisms multi kernel channel attention and spatial attention is used. These two attention mechanisms are used for obtaining spatial and channel attention feature maps. Finally, the classification process is carried out to classify the normal and diseased cases. The test accuracy results shows that our model surpasses the other models in terms of its classification performance.