原创力文档- 11
- 0
- 约2.3万字
- 约 18页
- 2024-02-28 发布于四川
- 举报
本发明提供了一种地铁车辆螺栓定位检测方法，首先制作地铁车辆螺栓数据集，通过改进YOLOv5网络对螺栓进行定位检测。针对螺栓像素占比小且骨干网络中多次卷积造成特征信息损失的问题，设计一种特征加权融合模块用于融合骨干网络提取的各尺度特征图，在融合时加入坐标注意力机制，为不同尺度特征图赋予权重，提高模型对螺栓的关注度；在主干网络中增加上下文信息增强模块，通过融合不同尺度感受野的特征，获得关于螺栓更加丰富的上下文信息，提高螺栓的定位精度；用NWD‑NMS替换YOLOv5网络对小尺度目标敏感的IOU‑NM
(19)国家知识产权局
(12)发明专利申请
(10)申请公布号CN117611967A
(43)申请公布日2024.02.27
(21)申请号202311469582.3G06N3/0464(2023.01)
文档评论（0）