ISSN 1009-5624 CN 10-2021/TQ 主管:中国乐凯集团有限公司 主办:北京乐凯科技有限公司
【摘要】针对工业机器人工件检测场景中目标检测模型计算资源受限与检测精度需求高的矛盾,本文提出一种基于 VoVNet 优化的轻量级单步多框检测算法( SSD)。 采用 VoVNet 替换 SSD 的主干网络 VGG16,通过单次聚合(OSA) 模块提升特征提取效率;在VoVNet 中引入残差结构并设计增强的残差单次聚合模块(residual one shot aggregation,Res⁃OSA)模块,解决网络退化问题并增强关键特征表达能力;进一步在 Res⁃OSA 模块后增加混合域注意力机制(HDA)提高定位精度。 实验表明:改进后的算法在 PASCAL VOC 数据集上平均精度(mAP)达到 79. 4%,每秒传输帧数(FPS)提升至 65,模型大小为 25. 8 MB,显著优于原 SSD 算法。