机床与液压

作者：付强，朱传军，梁泽启

单位：湖北工业大学机械工程学院；湖北工业大学机械工程学院；湖北工业大学机械工程学院

关键词：钢材表面缺陷;小目标检测;YOLOv5s;轻量化;注意力机制

针对传统带钢表面缺陷检测技术落后、效率不高及小目标识别能力不足等问题，提出一种改进的YOLOv5s-Tiny目标检测模型，在保持模型较小计算量的同时提升检测速度和识别精度。通过将主干网络GSP-Darknet53替换为轻量级GhostNet网络，减少模型参数的数量，提高推理速度。在主干网络加入CBAM注意力机制，通过通道注意力机制和空间注意力机制对特征信息进行融合增强，提高小目标检测精度，并将损失函数GIoU改进为EIoU，提高检测框定位能力。最后将改善后的训练模型格式转换后安装到手机安卓端验证优化的有效性。结果表明：在东北大学数据集中，改进后模型检测精度提高1.5%的同时，召回率提升了1.5%，参数量减少12.3%；安卓端检测速度约为120 ms，完成带钢缺陷的实时检测。

来源：2024年第10期

《机床与液压》期刊编辑部

查看机床与液压杂志2024年第10期