一种地铁车辆螺栓定位检测方法

本发明提供了一种地铁车辆螺栓定位检测方法，首先制作地铁车辆螺栓数据集，通过改进YOLOv5网络对螺栓进行定位检测。针对螺栓像素占比小且骨干网络中多次卷积造成特征信息损失的问题，设计一种特征加权融合模块用于融合骨干网络提取的各尺度特征图，在融合时加入坐标注意力机制，为不同尺度特征图赋予权重，提高模型对螺栓的关注度；在主干网络中增加上下文信息增强模块，通过融合不同尺度感受野的特征，获得关于螺栓更加丰富的上下文信息，提高螺栓的定位精度；用NWD‑NMS替换YOLOv5网络对小尺度目标敏感的IOU‑NMS，去除重叠的预测边界框。本发明有效增强了螺栓的特征表达能力，实现了更准确的螺栓定位检测。