YOLO-Wheat: A Wheat Disease Detection Algorithm Improved by YOLOv8s

被引:0
作者
Yao, Xiaotong [1 ]
Yang, Feng [1 ]
Yao, Jiayin [2 ]
机构
[1] Lanzhou Jiaotong Univ, Sch Elect & Informat Engn, Lanzhou 730070, Peoples R China
[2] Univ Glasgow, Sch Comp Sci, Gilmorehill Campus, Glasgow G12 8QQ, Scotland
来源
IEEE ACCESS | 2024年 / 12卷
关键词
Diseases; Feature extraction; Crops; Accuracy; Training; Convolutional neural networks; Detection algorithms; Tiny machine learning; C2f-DCN; disease detection; tiny target detection layer; wheat; YOLO-wheat;
D O I
10.1109/ACCESS.2024.3460394
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The objective of this research is to create an algorithm for identifying wheat diseases in their natural habitat. With its adaptability to small targets and intricate surroundings, the algorithm is expected to offer a precise foundation for scientific management and disease prevention. First, a total of 3622 original datasets of wheat disease photos were acquired by taking wheat photographs in an agricultural environment over various periods. Second, YOLO-Wheat, a detection method, is suggested for the properties of infected wheat ears and leaves that share a limited detection target area and comparable look, texture, color, and other aspects. In order to improve the extraction of illness characteristics and get a wider sensory field for the input features, the algorithm makes use of the new C2f-DCN module and the SCNet attention mechanism. Moreover, it enhances the model's capacity to extract information from distorted objects by learning the offset and weighting. Moreover, to improve the identification of minor illnesses, the design has enlarged the detection layer for tiny targets, modified the detecting head, and optimized the loss function. All of these changes have improved the accuracy of minor disease detection. On the experimental dataset, the YOLO-Wheat method earned a mAP@0.5 of 93.28%, which is 12% better than the original model. The suggested approach shows a 47% performance gain over the previous model, while maintaining a 23.94 MB smaller algorithm size. The results of this study show that the approach may greatly increase the model's capacity to extract features from tiny target pictures as well as the robustness of crop disease detection. As a result, the technique may be precisely and successfully used in real-world crop disease detection settings.
引用
收藏
页码:133877 / 133888
页数:12
相关论文
共 24 条
  • [1] Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks
    Chen, Jierun
    Kao, Shiu-Hong
    He, Hao
    Zhuo, Weipeng
    Wen, Song
    Lee, Chul-Ho
    Chan, S. -H. Gary
    [J]. 2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 12021 - 12031
  • [2] Deformable Convolutional Networks
    Dai, Jifeng
    Qi, Haozhi
    Xiong, Yuwen
    Li, Yi
    Zhang, Guodong
    Hu, Han
    Wei, Yichen
    [J]. 2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, : 764 - 773
  • [3] Hyperspectral Image Classification with Capsule Network Using Limited Training Samples
    Deng, Fei
    Pu, Shengliang
    Chen, Xuehong
    Shi, Yusheng
    Yuan, Ting
    Pu, Shengyan
    [J]. SENSORS, 2018, 18 (09)
  • [4] PDDD-PreTrain: A Series of Commonly Used Pre-Trained Models Support Image-Based Plant Disease Diagnosis
    Dong, Xinyu
    Wang, Qi
    Huang, Qianding
    Ge, Qinglong
    Zhao, Kejun
    Wu, Xingcai
    Wu, Xue
    Lei, Liang
    Hao, Gefei
    [J]. PLANT PHENOMICS, 2023, 5
  • [5] Gevorgyan Z, 2022, Arxiv, DOI [arXiv:2205.12740, DOI 10.48550/ARXIV.2205.12740]
  • [6] Hang Z., 2018, J. Shandong Agricult. Sci. China, V50, P137
  • [7] Hou QB, 2021, Arxiv, DOI [arXiv:2103.02907, 10.48550/arXiv.2103.02907, DOI 10.48550/ARXIV.2103.02907]
  • [8] Knowledge Distillation Facilitates the Lightweight and Efficient Plant Diseases Detection Model
    Huang, Qianding
    Wu, Xingcai
    Wang, Qi
    Dong, Xinyu
    Qin, Yongbin
    Wu, Xue
    Gao, Yangyang
    Hao, Gefei
    [J]. PLANT PHENOMICS, 2023, 5
  • [9] Jia S., 2019, Trans. Chinese Soc. Agric. Machin., V50, P313
  • [10] Recognition of rice leaf diseases and wheat leaf diseases based on multi-task deep transfer learning
    Jiang, Zhencun
    Dong, Zhengxin
    Jiang, Wenping
    Yang, Yuze
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2021, 186