YOLOV5-CBAM-C3TR: an optimized model based on transformer module and attention mechanism for apple leaf disease detection

被引:11
|
作者
Lv, Meng [1 ]
Su, Wen-Hao [1 ]
机构
[1] China Agr Univ, Coll Engn, Beijing, Peoples R China
来源
基金
中国国家自然科学基金;
关键词
deep learning; apple leaf; disease detection; YOLOv5; attention mechanism; transformer encoder;
D O I
10.3389/fpls.2023.1323301
中图分类号
Q94 [植物学];
学科分类号
071001 ;
摘要
Apple trees face various challenges during cultivation. Apple leaves, as the key part of the apple tree for photosynthesis, occupy most of the area of the tree. Diseases of the leaves can hinder the healthy growth of trees and cause huge economic losses to fruit growers. The prerequisite for precise control of apple leaf diseases is the timely and accurate detection of different diseases on apple leaves. Traditional methods relying on manual detection have problems such as limited accuracy and slow speed. In this study, both the attention mechanism and the module containing the transformer encoder were innovatively introduced into YOLOV5, resulting in YOLOV5-CBAM-C3TR for apple leaf disease detection. The datasets used in this experiment were uniformly RGB images. To better evaluate the effectiveness of YOLOV5-CBAM-C3TR, the model was compared with different target detection models such as SSD, YOLOV3, YOLOV4, and YOLOV5. The results showed that YOLOV5-CBAM-C3TR achieved mAP@0.5, precision, and recall of 73.4%, 70.9%, and 69.5% for three apple leaf diseases including Alternaria blotch, Grey spot, and Rust. Compared with the original model YOLOV5, the mAP 0.5increased by 8.25% with a small change in the number of parameters. In addition, YOLOV5-CBAM-C3TR can achieve an average accuracy of 92.4% in detecting 208 randomly selected apple leaf disease samples. Notably, YOLOV5-CBAM-C3TR achieved 93.1% and 89.6% accuracy in detecting two very similar diseases including Alternaria Blotch and Grey Spot, respectively. The YOLOV5-CBAM-C3TR model proposed in this paper has been applied to the detection of apple leaf diseases for the first time, and also showed strong recognition ability in identifying similar diseases, which is expected to promote the further development of disease detection technology.
引用
收藏
页数:13
相关论文
共 22 条
  • [21] A visual defect detection for optics lens based on the YOLOv5-C3CA-SPPF network model
    Tang, Haiyang
    Liang, Shan
    Yao, Dan
    Qiao, Yongjie
    OPTICS EXPRESS, 2023, 31 (02) : 2628 - 2643
  • [22] Improved YOLOv5m model based on Swin Transformer, K-means++, and Efficient Intersection over Union (EIoU) loss function for cocoa tree ( Theobroma cacao) ) disease detection
    Nyarko, Benedicta Nana Esi
    Bin Wu, Wu
    Zhou, Zhou Jinzhi
    Mohd, Mwanaharusi Mohd Juma
    JOURNAL OF PLANT PROTECTION RESEARCH, 2024,