PSOS-YOLOv5s:一种轻量级玉米雄穗快速检测算法

打开文本图片集
摘 要:针对玉米雄穗检测中速度较慢的问题,提出一种基于YOLOv5s 改进的轻量化快速检测算法———PSOS-YOLOv5s。所提算法在主干网络中采用轻量级PP-LCNet 替换CSPDarknet53,采用2 种不同的深度可分离卷积与注意力机制的组合来构成基本块,降低模型复杂度并加快检测速度;在预测部分采用SimOTA 标签匹配策略替换YOLOv5s 中的标签匹配策略,采用中心先验思想获得精准的先验知识,提出动态k 策略过滤冗余标签,提高模型对正样本的快速选取能力;在预测部分采用SIOULoss 替换GIOU Loss,引入角度损失因子来降低回归自由度、加快收敛速度、节省训练时间,重新定义惩罚指标,提高检测精度。(剩余13343字)