基于改进YOLO v5的轻量化苹果检测方法

扫码查看

原文链接

NETL
NSTL
万方数据

中文摘要：为了实现苹果采摘过程中准确快速的识别,本研究提出一种融合FasterNet模型的YOLO v5改进苹果检测算法.首先在基准图像特征提取模块中使用FasterNet架构替代YOLO v5模型中的卷积块和CSPLayer,降低算法复杂度并增强小目标的特征提取能力;然后提出了利用ECIoU损失函数来预测目标位置偏差,通过增加边框角点损失来描述预测框与真实目标框之间的位置偏差信息,进一步提高了苹果检测的准确性,解决了 YOLO v5算法对有遮挡的密集目标检测效果不佳的问题;最后在检测后处理阶段提出ECIoU-NMS方法以优化重叠目标框的选择.在通用数据集MS COCO和自建数据集上对本研究所提方法与YOLO v5算法进行了对比试验.本研究所提算法模型参数量下降了12％,计算量下降了25％,帧率提升了23％.在通用数据集MS COCO上m.AP0.5和mAP0 5∶0.95指标分别提升了2.9、1.9百分点,在自建苹果数据集上mAP0.5和mAP0.5∶0.95指标分别提升了2.5、1.9百分点.本研究方法性能优于YOLO v5,且模型的轻量化使其更容易在苹果采摘机器人上部署.

作者：

温彬彬、张华、孟祥龙

展开 >

作者单位：

河北工业职业技术大学,河北石家庄,054000

河北农业大学,河北保定,071000

关键词：

目标检测 YOLO v5 FasterNet网络边框角点 ECIoU

基金：

河北省自然科学基金石家庄市科技计划河北省"三三三人才工程"项目

项目编号：

C2021204034231130351A202101035

出版年：

2024

DOI：

10.15889/j.issn.1002-1302.2024.12.029

江苏农业科学

江苏省农业科学院

江苏农业科学

CSTPCD北大核心

影响因子：0.732

ISSN：1002-1302

年,卷(期)：2024.52(12)