基于改进YOLOv5s的桥梁螺栓缺陷识别方法 |
| |
引用本文: | 张洪,朱志伟,胡天宇,龚燕峰,周建庭.基于改进YOLOv5s的桥梁螺栓缺陷识别方法[J].吉林大学学报(工学版),2024(3):749-760. |
| |
作者姓名: | 张洪 朱志伟 胡天宇 龚燕峰 周建庭 |
| |
作者单位: | 1. 重庆交通大学省部共建山区桥梁及隧道工程国家重点实验室;2. 重庆交通大学信息科学与工程学院;3. 重庆交通大学航运与船舶工程学院 |
| |
基金项目: | 国家自然科学基金项目(52278291,U20A20314);;重庆市自然科学基金项目(CSTB2022NSCQ-LZX0006,CSTB2022TIAD-KPX0205); |
| |
摘 要: | 针对现有算法在检测桥梁螺栓缺陷时因螺栓背景复杂和尺寸较小而导致的特征提取不充分、目标定位不精确问题,提出了一种基于改进YOLOv5s的桥梁螺栓缺陷识别方法。该方法在骨干网络中引入注意力机制以提升模型对螺栓特征的提取能力并加深对螺栓全局特征的关注度;优化空间金字塔池化结构以减少螺栓特征信息流失;采用MPDIoU作为边界框回归损失函数,提高螺栓边界框的回归精度;将YOLO检测头解耦以消除目标检测中分类任务和回归任务共享检测头对边界框位置回归的负面影响。在螺栓锈蚀、螺栓松动、螺栓脱落和螺母脱落4类典型缺陷螺栓以及正常螺栓的3810张自制螺栓图像数据集上进行训练和测试,实验结果表明:本文算法对螺栓缺陷的检测精度达到90.8%,相较于YOLOv5s提升了3%,均值平均精度达到92.6%,相较于YOLOv5s提升了4.3%,可以应用于桥梁螺栓的缺陷智能识别。
|
关 键 词: | 桥梁工程 螺栓缺陷识别 YOLOv5s 桥梁螺栓 |
|
|