陈小康副教授等:基于改进YOLOv5s的复杂环境下新梅检测方法

阅读文章全文：http://www.tcsae.org/article/doi/10.11975/j.issn.1002-6819.202403060

《农业工程学报》2024年第40卷第14期刊载了新疆大学等单位董耿耿、陈小康、樊湘鹏、周建平、姜宏与崔超的论文——“基于改进YOLOv5s的复杂环境下新梅检测方法”。该研究由自治区科技计划项目-天山创新团队计划项目（项目号：2022D14002）资助。

引文信息：董耿耿，陈小康，樊湘鹏，等. 基于改进YOLOv5s的复杂环境下新梅检测方法[J]. 农业工程学报，2024，40(14)：118-125.

DOI: 10.11975/j.issn.1002-6819.202403060

研究目的与方法：

为解决新梅在树干树叶遮挡、果实重叠情况下难以准确检测的问题，该研究建立了新梅目标检测模型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个百分点，能够满足果园复杂环境下对新梅进行实时检测的需求，为后续新梅采摘机器人的视觉感知环节提供了技术支持。