复杂场景下害虫目标检测算法：YOLOv8-Extend

分类： 农、林、牧、渔 >> 农、林、牧、渔业科学其他学科 提交时间： 2024-06-17 合作期刊: 《智慧农业（中英文）》

张荣华 白雪 樊江川

摘要： ［目的/意义］实现复杂的自然环境下农作物害虫的识别检测，改变当前农业生产过程中依赖于专家人工 感官识别判定的现状，提升害虫检测效率和准确率具有重要意义。针对农作物害虫目标检测具有目标小、与农作 物拟态、检测准确率低、算法推理速度慢等问题，本研究提出一种基于改进YOLOv8的复杂场景下农作物害虫目 标检测算法。［方法］首先通过引入GSConv提高模型的感受野，部分Conv更换为轻量化的幻影卷积（Ghost Convo⁃ lution），采用HorBlock捕捉更长期的特征依赖关系，Concat更换为BiFPN（Bi-directional Feature Pyramid Network） 更加丰富的特征融合，使用VoVGSCSP 模块提升微小目标检测，同时引入CBAM （Convolutional Block Attention Module） 注意力机制来强化田间虫害目标特征。然后使用Wise-IoU损失函数更多地关注普通质量样本，提高网络 模型的泛化能力和整体性能。之后，对改进后的YOLOv8-Extend 模型与YOLOv8 原模型、YOLOv5、YOLOv8- GSCONV、YOLOv8-BiFPN、YOLOv8-CBAM进行对比，验证模型检测准确度和精度。最后将模型移植到边缘设备 进行推理验证，在实际应用场景中验证模型的有效性。［结果和讨论］YOLOv8-Extend在对比实验中均取得良好的 表现，其中与原模型对比实验中，精确率、召回率、mAP@0.5和mAP@0.5∶0.95评价指标分别提升2.6%、3.6%、 2.4% 和7.2%，表现突出，具有更好的检测效果。改进前后的模型分别运行在边缘计算设备JETSON ORIN NX 16 GB上并通过TensorRT加速相比，mAP@0.5提升4.6%，达到57.6 FPS，满足实时性检测要求。在复杂农业场景 中YOLOv8-Extend模型具有更好的适应性，在实际采集数据中微小害虫与生长环境相似的害虫检测方面有明显优 势，在困难数据检测方面准确率提高了11.9%。［结论］本研究提出的YOLOv8改进模型有效提高了检测精度和识 别率同时保持了较高的运行效率，能够部署在边缘终端计算设备上实现农作物害虫的实时检测，也为其他小目标 智能检测和模型结构优化提供参考和帮助。

通过

 点击量 139  下载量 43  评论 0