高级检索+

基于YOLO-EDW的农林害虫检测方法

Agricultural and forestry pest detection method based on YOLO-EDW

  • 摘要: 为解决农林害虫图像存在不同尺度目标、相近科属外观相似导致的模型检测精度不佳问题,该研究提出一种基于YOLOv8n的农林害虫检测模型YOLO-EDW。采用注意力机制EMA(efficient multi-scale attention)强化关键特征提取能力;引入检测头DBB(diverse branch block),借助多分支结构对近科属害虫的特征进行区分;利用小波变换卷积对C2f(cross-stage partial bottleneck with two convolutions)结构进行优化实现特征融合,从而聚焦于农林害虫纹理和形状信息。在包含71类农林害虫数据集上试验结果表明,YOLO-EDW模型精确度、召回率和平均精度均值分别达到了90.6%、88.6%和92.0%,相比于YOLOv8n模型分别提高了2.3、1.2和0.4个百分点。与主流目标检测SSD和Faster R-CNN等模型进行对比,YOLO-EDW模型平均精度均值分别提高8.4和3.5个百分点。YOLO-EDW模型能够实现对不同尺度目标、相近科属害虫有效检测,并可为农林害虫实际智能监测提供技术支撑。

     

    Abstract: Image-based detection has been widely applied in agricultural and forestry pest monitoring. However, challenges remain in real-world environments, such as serious size variations and visual similarities between pests from closely related families, leading to low detection accuracy. In this study, an improved model was proposed to detect pests in agriculture and forestry using YOLO-EDW. Initially, YOLOv8n was used as the baseline model to tackle the missed detections due to small targets. An efficient multi-scale attention mechanism was incorporated to improve the extraction of key features, along with a cross-space learning approach. Fusion features were generated to capture both short- and long-range dependencies using multi-scale parallel sub-networks, effectively extracting pest features. A diverse branch block detection head was introduced to reduce the misdetections caused by pests with similar appearances. This multi-branch structure increased the complexity of convolutional layers during training. High performance was achieved in the differentiate pests from closely related families. Additionally, wavelet transform convolution was applied to optimize the C2f structure for feature fusion, where the texture and shape information were extracted from forestry pests. Fine-grained details were captured under consistent experimental conditions. Ablation tests were conducted on the AFPD (agricultural and forestry pest dataset), including 71 species of agricultural and forestry pests. The results revealed that the YOLO-EDW model achieved an accuracy, recall, and mean average precision of 90.6%, 88.6%, and 92.0%, respectively, which were improved by 2.3,1.2, and 0.4 percentage points, compared with the YOLOv8n model. The YOLO-EDW model’s mAP was improved by 8.4 and 3.5 percentage points, respectively, compared with popular object models, like SSD and Faster R-CNN. Visualization results confirmed that the high effectiveness was obtained for detecting agricultural and forestry pests. A comparison was also made between the YOLO-EDW and YOLOv8n models using pest images of varying target sizes from the AFPD. The YOLO-EDW model with a feature enhancement module accurately detected both small and occluded pests under complex scenarios. YOLOv8n suffered a higher misdetection rate for pests like the Cnaphalocrocis medinalis. As such, the YOLO-EDW model effectively distinguished between morphologically similar pests, better supporting the monitoring needs of forestry pests. Furthermore, the generalization was tested on the publicly available IP102 dataset. The improved YOLOv8n model outperformed mainstream object algorithms, like Faster R-CNN, in mAP. Highly precise localization and classification were observed on both adult and larval pests in crops, indicating effective detection of small pests, like aphids in clusters, the normally missed and misdetected in dense target scenarios. Overall, the improved model adapted well to the morphological and distribution features of pests in agricultural and forestry production, fully meeting the practical demands for high accuracy and robustness. Therefore, the YOLO-EDW model with only 5.58×106 parameters, high inference speed, and easy deployment provides high accuracy for pests at varying scales and families.

     

/

返回文章
返回