一种基于改进YOLOv5的荔枝果实生长期识别方法

    公开(公告)号:CN115984698A

    公开(公告)日:2023-04-18

    申请号:CN202310010754.4

    申请日:2023-01-05

    Abstract: 本发明公开了一种基于改进YOLOv5的荔枝果实生长期识别方法,包括:1)荔枝数据集的建立,包括对数据采集、预处理、标注和划分数据集;2)对传统YOLOv5的特征提取模块、特征融合模块和下采样方式进行改进。在特征提取模块的SPPF模块后添加SimAM注意力机制。在特征融合模块中,使用轻量级卷积GSConv替代普通卷积,既保证了检测精度与速度,也降低了计算成本;将YOLOv5网络中的所有3×3的跨步卷积替换为非跨步卷积,并在非跨步卷积之后添加SPD模块用于下采样;最终由预测模块对输出的数据进行信息分析和处理,得出检测结果。本发明解决了荔枝果实生长期识别中,果实目标小、重叠、遮挡和图像模糊导致目标检测精度低的问题。

Patent Agency Ranking