农业机械学报2024,Vol.55Issue(11):68-74,7.DOI:10.6041/j.issn.1000-1298.2024.11.007
基于改进YOLO v7的苹果叶片病害检测方法
Apple Leaf Disease Detection Method Based on Improved YOLO v7
摘要
Abstract
Apples have become one of the most popular fruits in the world,and the annual production of apples in China has continued to increase.However,there are certain diseases in the growth process of apple trees,which will affect the quality and yield of apples,resulting in economic losses of fruit farmers.Therefore,in view of the problem that apple leaf diseases have diverse forms and dense distribution,resulting in low detection accuracy,an improved YOLO v7 model was proposed to accurately detect apple leaf diseases.Firstly,bidirectional feature pyramid network(BiFPN)was used to replace the original feature fusion method in YOLO v7 to improve the model's detection ability of different scale diseases on apple leaves.Secondly,after the ELAN and E-ELAN modules of YOLO v7,an efficient channel attention mechanism(EC A)was added to enhance the ability of the model to extract features of apple leaves disease and improve detection accuracy.Finally,the loss function of YOLO v7 was changed to the SIOU loss function to accelerate the convergence speed of the model.Experimental results showed that the improved YOLO v7 model had a precision of 89.4%,a recall rate of 81.5%,a mean average precision(mAP@0.5)of 90.5%,and a mean average precision(mAP@0.95)of 62.1%.Compared with the original YOLO v7 model,they were increased by 4.9,5.2,3.5,and 4.6 percentage points,respectively.Compared with the Faster R-CNN,SSD,YOLO v3,YOLO v5s,and YOLO v7 models,the mAP@0.5 of improved YOLO v7 model was increased by 40.9,20.3,4.0,2.3 and 3.5 percentage points,respectively,and the single image detection speed reached 12 ms.The research can provide a feasible technical means for accurately detecting apple leaf diseases.关键词
苹果叶片/病害检测/YOLO v7/多尺度融合/注意力机制Key words
apple leaf/disease detection/YOLO v7/multi-scale fusion/attention mechanism分类
信息技术与安全科学引用本文复制引用
袁杰,谢霖伟,郭旭,梁荣光,张迎港,马浩田..基于改进YOLO v7的苹果叶片病害检测方法[J].农业机械学报,2024,55(11):68-74,7.基金项目
国家自然科学基金项目(62263031)和新疆维吾尔自治区自然科学基金项目(2022D01C53) (62263031)