基于轻量化网络与嵌入式系统的喷码检测 |
| |
作者姓名: | 葛俏 梁桥康 邹坤霖 孙炜 李珊红 王耀南 |
| |
作者单位: | 1. 湖南大学电气与信息工程学院;2. 湖南大学机器人视觉感知与控制技术国家工程研究中心;3. 合肥学院先进制造工程学院 |
| |
基金项目: | 国家自然科学基金资助项目(62073129,U21A20490);;湖南省自然科学基金资助项目(2022JJ10020);;安徽省自然科学基金资助项目(1808085QF195); |
| |
摘 要: | 针对食品饮料等复杂包装上喷码质量检测的准确率不高与速度慢等问题,提出了一种基于Ghost-YOLO轻量化网络与嵌入式平台的喷码质量检测方法。网络以YOLOv5为基础,采用了幻影模块(GM)对卷积层进行降维,模型参数减少25%。多分类目标检测任务的后处理采用位置重复抑制(PDS)方法,通过对所有类别同时采用非极大值抑制(NMS),进一步提高检测精度。最后,利用所提出的改进自训练方法对模型进行训练,并将所提检测方法部署于嵌入式设备中,实现了对喷码质量的实时检测。实验结果表明,所提检测方法在满足实时性的要求下,对喷码字符检测的精确度和召回率分别达到了100%和99.99%。
|
关 键 词: | YOLOv5 幻影模块 目标检测 非极大值抑制 自训练 Jetson TX2 |
|