Cotton leaf disease detection method based on improved SSD

Wenjuan Guo, Shuo Feng, Quan Feng, Xiangzhou Li, Xueze Gao

Abstract


In response to the problems of numerous model parameters and low detection accuracy in SSD-based cotton leaf disease detection methods, a cotton leaf disease detection method based on improved SSD was proposed by combining the characteristics of cotton leaf diseases. First, the lightweight network MobileNetV2 was introduced to improve the backbone feature extraction network, which provides more abundant semantic information and details while significantly reducing the amount of model parameters and computing complexity, and accelerates the detection speed to achieve real-time detection. Then, the SE attention mechanism, ECA attention mechanism, and CBAM attention mechanism were fused to filter out disease target features and effectively suppress the feature information of jamming targets, generating feature maps with strong semantics and precise location information. The test results on the self-built cotton leaf disease dataset show that the parameter quantity of the SSD_MobileNetV2 model with backbone network of MobileNetV2 was 50.9% of the SSD_VGG model taking VGG as the backbone. Compared with SSD_VGG model, the P, R, F1 values, and mAP of the MobileNetV2 model increased by 4.37%, 3.3%, 3.8%, and 8.79% respectively, while FPS increased by 22.5 frames/s. The SE, ECA, and CBAM attention mechanisms were introduced into the SSD_VGG model and SSD_MobileNetV2 model. Using gradient weighted class activation mapping algorithm to explain the model detection process and visually compare the detection results of each model. The results indicate that the P, R, F1 values, mAP and FPS of the SSD_MobileNetV2+ECA model were higher than other models that introduced the attention mechanisms. Moreover, this model has less parameter with faster running speed, and is more suitable for detecting cotton diseases in complex environments, showing the best detection effect. Therefore, the improved SSD_MobileNetV2+ECA model significantly enhanced the semantic information of the shallow feature map of the model, and has a good detection effect on cotton leaf diseases in complex environments. The research can provide a lightweight, real-time, and accurate solution for detecting of cotton diseases in complex environments.
Key words: cotton disease detection; SSD; MobileNetV2; attention mechanism
DOI: 10.25165/j.ijabe.20241702.8574

Citation: Guo W J, Feng S, Feng Q, Li X Z, Gao X Z. Cotton leaf disease detection method based on improved SSD. Int J
Agric & Biol Eng, 2024; 17(2): 211–220.

Full Text:

PDF

References


Guo W J, Feng Q, Li X Z, Yang S, Yang J Q. Grape leaf disease detection based on attention mechanisms. Int J Agric & Biol Eng, 2022; 15(5): 205–212.

Wang J W, Zhao L H, Shi Y Q, Wei F, Feng H J, Zhu H Q, et al. Preliminary report on the whole process control techniques of cotton diseases. China Cotton, 2020; 47(5): 20–22, 46. (in Chinese)

Zhang J H, Ji R H, Yuan X, Li H, Qi L J. Recognition of pest damage for cotton leaf based on RBF-SVM algorithm. Transactions of the CSAM, 2011; 42(8): 178–183. (in Chinese)

Zhang J H, Qi L J, Ji R H, Wang H, Huang S K, Wang P. Cotton diseases identification based on rough sets and BP neural network. Transactions of the CSAE, 2012; 28(7): 161–167. (in Chinese)

Zhang J H, Han S Q, Zhai Z F, Kong F T, Feng X, Wu J Z. Improved adaptive watershed method for segmentation of cotton leaf adhesion lesions. Transactions of the CSAE, 2018; 34(24): 165–174. (in Chinese)

Zhai Z F, Xu Z, Zhou X Q, Wang L L, Zhang J H. Recognition of hazard grade for cotton blind stinkbug based on Naive Bayesian classifier. Transactions of the CSAE, 2015; 31(1): 204–211. (in Chinese)

Nazki H, Yoon S, Fuentes A, Dong S P. Unsupervised image translation using adversarial networks for improved plant disease recognition. Computers and Electronics in Agriculture, 2020; 168: 105–117.

Liu Y, Gao G Q. Identification of multiple leaf diseases using improved SqueezeNet model. Transactions of the CSAE, 2021; 37(2): 187–195. (in Chinese)

Wang C S, Zhao C J, Wu H R, Zhou J, Li J X, Zhu H J. Recognizing crop diseases using bimodal joint representation learning. Transactions of the CSAE, 2021; 37(11): 180–188. (in Chinese)

Li S Q, Chen C, Zhu T, Liu B. Plant leaf disease identification based on lightweight residual network. Transactions of the CSAM, 2022; 53(3): 243–250. (in Chinese)

Zhang J H, Kong F T, Wu J Z, Zhai Z F, Han S Q, Cao S S. Cotton disease recognition model based on improved VGG convolutional neural network. Journal of China Agricultural University, 2018; 23(11): 167–177. (in Chinese)

Wang X F, Zhang C L, Zhang S W, Zhu Y H. Forecasting of cotton diseases and pests based on adaptive discriminant deep belief network. Transactions of the CSAE, 2018; 34(14): 157–164. (in Chinese)

Zhao L X, Hou F D, Lyu Z C, Zhu H C, Ding X L. Image recognition of cotton leaf diseases and pests based on transfer learning. Transactions of the CSAE, 2020; 36(7): 184–191. (in Chinese)

Chen K Q, Zhu Z L, Deng X M, Ma C X, Wang H A. Deep learning for multi-scale object detection: A survey. Journal of Software, 2021; 32(4): 1201–1227. (in Chinese)

Zhou H R, Wu B M. Advances in research on deep learning for crop disease image recognition. Journal of Agricultural Science and Technology, 2021; 23(5): 61–68. (in Chinese)

Fuentes A, Yoon S, Kim S C, Park D S. A robust deep-learning-based detector for real-time tomato plant diseases and pests recognition. Sensors 2017; 17(9): 2022. doi: 10.3390/s17092022.

Li J H, Lin L J, Tian K, Alaa A A. Detection of leaf diseases of balsam pear in the field based on improved Faster R-CNN. Transactions of the CSAE, 2020; 36(12): 179–185. (in Chinese)

Ye Z H, Zhao M X, Jia L. Image recognition of crop diseases in complex background. Transactions of the CSAM, 2021; 52(S0): 118–124,147. (in Chinese)

Liu J, Wang X. Early recognition of tomato gray leaf spot disease based on MobileNetv2-YOLOv3 model. Plant Methods, 2020; 16: 83.

Wen B, Cao R X, Yang Q L, Zhang J, Zhu H, Li Z C. Detecting leaf disease for Panax notoginseng using an improved YOLOv3 algorithm. Transactions of the CSAE, 2022; 38(3): 164–172. (in Chinese)

Ren S Q, He K M, Girshick R, Sun J. Faster R-CNN: Towards real-time object detection with region proposal networks. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017; 39(6): 1137–1149.

Redmon J, Divvala S, Girshick R, Farhadi A. You only look once: Unified, real-time object detection. IEEE Conference on Computer Vision and Pattern Recognition, 2016; pp.779–788.

Redmon J, Farhadi A. YOLOv3: An incremental improvement. (2018-04-08).https://arxiv.org/pdf/1804. 02767. pdf.

Bochkovskiy A, Wang C Y, Liao H. YOLOv4: Optimal speed and accuracy of object detection. (2020-04-23). https://arxiv.or/abs/2004. 10934.

Zheng G, Liu S T, Wang F, Li Z M, Sun J. YOLOx: Exceeding YOLO series in 2021. arXiv preprint, 2107; 08430.

Liu W, Anguelov D, Erhan D, Szegedy C, Reed S, Fu C Y, et al. SSD: Single shot multibox detector. European Conference on Computer Vision, Springer, Cham, 2016; pp.21–37.

Simonyan K, Zisserman A. Very deep convolutional networks for large-scale image recognition. arXiv preprint, 2015; pp.1409–1556.

Sandler M, Howard A, Zhu M L, Zhmoginov A, Chen L. MobileNetV2: inverted residuals and linear bottlenecks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2018; pp.4510-4520.

Dai F, Li X Z, Shi R J, Zhang F W, Zhao W Y, Guo W J. Film identification method based on improved deeplabv3+ for full-film double-ditch corn seedbed. Int J Agric & Biol Eng, 2023; 16(5): 165–172.

Hu J, Li S, Sun G. Squeeze-and-excitation networks. The IEEE conference on computer vision and pattern recognition. Piscataway, New York, USA: IEEE, 2018; pp.2011–2023.

Woo S, Park J, Lee J Y, Kweon I S. CBAM: Convolutional block attention module. European Conference on Computer Vision, Munish: Springer, 2018; pp.3–19.

Wang Q L, Wu B G, Zhu P F, Li P H, Zuo W M, Hu Q H. ECA-Net: Efficient channel attention for deep convolutional neural networks. IEEE/CVF Conference on Computer Vision and Pattern Recognition, Piscataway, NJ: IEEE Press, 2020.

Selvaraju R R, Cogswell M, Das A, Vedantam R, Parikh D, Batra D. Grad-CAM: visual explanations from deep networks via gradient-based localization. 2017 IEEE International Conference on Computer Vision (ICCV), Piscataway, NJ, USA, IEEE, 2017: pp.618–626.




Copyright (c) 2024 International Journal of Agricultural and Biological Engineering

Creative Commons License
This work is licensed under a Creative Commons Attribution 4.0 International License.

2023-2026 Copyright IJABE Editing and Publishing Office