文献类型： 中文期刊

作者： 董耿耿 ¹ ; 陈小康 ¹ ; 樊湘鹏 ² ; 周建平 ¹ ; 姜宏 ¹ ; 崔超 ¹ ;

作者机构： 1.新疆大学机械工程学院

2.中国农业科学院农业信息研究所

关键词： 图像处理;模型;新梅检测;YOLOv5s;注意力机制;双向特征金字塔;深度学习

期刊名称： 农业工程学报

ISSN： 1002-6819

年卷期： 2024 年 40 卷 014 期

页码： 118-125

收录情况： EI ; 北大核心 ; CSCD

摘要： 为解决新梅在树干树叶遮挡、果实重叠情况下难以准确检测的问题,该研究建立了新梅目标检测模型SFF-YOLOv5s.在真实果园环境下构建新梅数据集,以YOLOv5s模型作为基础网络,首先在Backbone骨干网络C3模块中引入CA(coordinate attention)注意力机制以增强模型对新梅关键特征信息的提取能力并减少模型的参数量;其次在Neck层中引入加权双向特征金字塔网络,增强模型不同特征层之间的融合能力,从而提高模型的平均精度均值;最后使用SIoU损失函数替换原模型中的CIoU损失函数提高模型的检测准确率.试验结果表明,SSF-YOLOv5s模型对新梅检测准确率为93.4%,召回率为92.9%,平均精度均值为97.7%,模型权重仅为13.6MB,单幅图像平均检测时间12.1ms,与 Faster R-CNN、YOLOv3、YOLOv4、YOLOv5s、YOLOv7、YOLOv8s 检测模型相比平均精度均值分别提升了 3.6、6.8、13.1、0.6、0.4、0.5个百分点,能够满足果园复杂环境下对新梅进行实时检测的需求,为后续新梅采摘机器人的视觉感知环节提供了技术支持.