文献类型： 中文期刊

作者： 李丽 ¹ ; 卢世博 ¹ ; 任浩 ¹ ; 徐刚 ² ; 周永忠 ³ ;

作者机构： 1.西南大学工程技术学院

2.北京市农林科学院智能装备技术研究中心

3.重庆祥飞智能科技有限公司

关键词： 桑叶采摘;枝干识别定位;YOLO v5;目标检测;注意力机制;坐标转换

期刊名称： 农业机械学报

ISSN： 1000-1298

年卷期： 2024 年 002 期

页码： 249-257

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

摘要： 为实现复杂自然环境下对桑树嫩叶处枝干的识别检测，改变当前桑叶采摘设备作业过程中依赖人工辅助定位的现状，解决识别目标姿态多样和环境复杂导致的低识别率问题，提出一种基于改进YOLO v5模型的桑树枝干识别模型（YOLO v5-mulberry）,并结合深度相机构建定位系统。首先，在YOLO v5的骨干网络中加入CBAM（Convolutional block attention module）注意力机制，提高神经网络对桑树枝干的关注度；并增加小目标层使模型可检测4像素×4像素的目标，提高了模型检测小目标的性能；同时使用GIoU损失函数替换原始网络中的IoU损失函数，有效防止了预测框和真实框尺寸较小时无法正确反映预测框及真实框之间位置关系的情况；随后，完成深度图和彩色图的像素对齐，通过坐标系转换获取桑树枝干三维坐标。试验结果表明：YOLO v5-mulberry检测模型的平均精度均值为94.2%,较原模型提高16.9个百分点，置信度也提高12.1%;模型室外检测时应检测目标数53,实际检测目标数为48,检测率为90.57%;桑树嫩叶处枝干三维坐标识别定位系统的定位误差为（9.498 5 mm, 11.285 mm, 19.11 mm）,满足使用要求。该研究可实现桑树嫩叶处枝干的识别与定位，有助于推动桑叶智能化采摘机器人研究。