Please wait a minute...
浙江大学学报(工学版)  2026, Vol. 60 Issue (11): 2418-2430    DOI: 10.3785/j.issn.1008-973X.2026.11.011
计算机技术     
基于改进RT-DETR的接触网吊弦螺栓小目标故障检测算法
顾桂梅1(),张新龙1,赵克进2
1. 兰州交通大学 自动化与电气工程学院,甘肃 兰州 730070
2. 中国铁路兰州局集团有限公司 银川供电段,宁夏回族自治区 银川 750000
Small-target fault detection algorithm for dropper bolts in overhead catenary systems based on improved RT-DETR
Guimei GU1(),Xinlong ZHANG1,Kejin ZHAO2
1. School of Automation and Electrical Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
2. Yinchuan Power Supply Section, China Railway Lanzhou Group Limited Company, Yinchuan 750000, China
 全文: PDF(3556 KB)   HTML
摘要:

铁路接触网吊弦螺栓故障检测中存在目标尺寸小、特征提取困难的问题,为此提出基于改进RT-DETR的轻量级小目标检测算法. 将融合上下文引导模块的CG-EfficientFormerV2作为特征提取网络,在保持多尺度特征表征能力的同时降低模型复杂度. 通过集成DySample动态上采样策略和上下文引导模块下采样机制,实现采样点在目标区域的自适应分布,增强模型对螺栓细节特征的捕获能力. 引入MASAG注意力机制,通过空间选择性调制机制实现局部-全局上下文信息的有效融合. 设计Inner-MPDIoU联合损失函数,结合内部关键点距离度量和长宽比约束,优化小目标定位精度. 在自建数据集上的实验表明,所提算法的mAP@0.5和mAP@0.5:0.95分别达到97.2%和69.2%,较基准模型分别提升4.0和6.6个百分点,参数量降低23.4%,计算复杂度降低31.53%,帧率达到40.6帧/s,满足实时性要求. 在VisDrone2019数据集上的泛化实验中,所提算法的mAP@0.5和mAP@0.5:0.95分别达到48.6%和29.0%,较基准模型分别提升5.8和3.0个百分点,具有跨域泛化能力. 实验结果验证了所提算法的有效性,表明该算法能够满足接触网吊弦螺栓轻量化小目标检测的需求.

关键词: 接触网吊弦螺栓;  RT-DETR;  小目标故障检测;  轻量化;  上下文引导模块    
Abstract:

A lightweight small-target detection algorithm based on improved RT-DETR was proposed to address the challenges of small target size and difficult feature extraction in fault detection of railway overhead catenary dropper bolts. The CG-EfficientFormerV2, incorporating context-guided blocks, was adopted as the feature extraction network, maintaining multi-scale feature representation capabilities while reducing model complexity. The DySample dynamic upsampling strategy and the context-guided-block downsampling mechanism were integrated to achieve adaptive distribution of sampling points within target regions and enhance the model’s capability to capture detailed bolt features. The MASAG attention mechanism was introduced to enable effective fusion of local-global contextual information through a spatially selective modulation mechanism. An Inner-MPDIoU joint loss function was designed by combining internal keypoint distance metrics and aspect ratio constraints to optimize small-target localization accuracy. Experiments conducted on a custom-built dataset demonstrated that the proposed algorithm achieved 97.2% and 69.2% in terms of mAP@0.5 and mAP@0.5:0.95, respectively, with improvements of 4.0 and 6.6 percentage points over the baseline. Additionally, the number of parameters was reduced by 23.4%, computational complexity was decreased by 31.53%, and an FPS of 40.6 was achieved, meeting real-time requirements. In generalization experiments on the VisDrone2019 dataset, the proposed algorithm achieved 48.6% and 29.0% in terms of mAP@0.5 and mAP@0.5:0.95, respectively, with improvements of 5.8 and 3.0 percentage points over the baseline, thereby validating the cross-domain generalizability. Experimental results verified the effectiveness of the proposed algorithm, which meets the requirements for lightweight small-target detection of dropper bolts in overhead catenary systems.

Key words: dropper bolts in overhead catenary systems    RT-DETR    small-object fault detection    lightweight    context-guided block
收稿日期: 2025-09-14 出版日期: 2026-10-16
CLC:  TP 393  
基金资助: 国家自然科学基金资助项目(61661027);中国国家铁路集团有限公司基金资助项目(N2022G012).
作者简介: 顾桂梅(1970—),女,教授,从事接触网智能监测和故障监测诊断研究. orcid.org/0000-0003-0485-5535. E-mail:386509464@qq.com
服务  
把本文推荐给朋友
加入引用管理器
E-mail Alert
作者相关文章  
顾桂梅
张新龙
赵克进

引用本文:

顾桂梅,张新龙,赵克进. 基于改进RT-DETR的接触网吊弦螺栓小目标故障检测算法[J]. 浙江大学学报(工学版), 2026, 60(11): 2418-2430.

Guimei GU,Xinlong ZHANG,Kejin ZHAO. Small-target fault detection algorithm for dropper bolts in overhead catenary systems based on improved RT-DETR. Journal of ZheJiang University (Engineering Science), 2026, 60(11): 2418-2430.

链接本文:

https://www.zjujournals.com/eng/CN/10.3785/j.issn.1008-973X.2026.11.011    或    https://www.zjujournals.com/eng/CN/Y2026/V60/I11/2418

图 1  RT-DETR模型
图 2  改进的RT-DETR模型
图 3  上下文引导模块示意图
图 4  CG-EfficientFormerV2网络架构
图 5  DySample动态上采样过程
图 6  DySample中的采样生成器
图 7  空间-通道引导的多尺度注意力机制
图 8  Inner-MPDIou联合损失函数示意图
图 9  数据集预处理
图 10  自适应直方图均衡化算法
图 11  吊弦螺栓的正常和缺陷样本
EfficientFormerV2改进后的EfficientFormerV2DySampleMASAGCGBmAP@0.5/%mAP@0.5:0.95/%NP/106Flops/109Rf/(帧· s?1)
—————93.262.920.0929.3112.4
√————95.063.711.8015.9735.6
—√———95.365.512.1716.3939.1
—√√——95.665.212.0216.0139.7
—√—√—95.465.415.3620.5334.6
—√——√95.864.814.9218.9538.9
—√√√—96.368.014.4119.3736.4
—√—√√97.069.213.6317.6933.8
—√√—√96.867.612.3716.8437.9
—√√√√97.269.515.3820.0740.6
表 1  改进模块消融实验结果
算法mAP@0.5/%mAP@0.5:0.95/%Flops/109NP/106Rf/(帧· s?1)
YOLOv5m97.469.832.3025.0413.4
YOLOv5s93.161.012.109.1538.9
YOLOv8m97.269.639.6625.8412.1
YOLOv8s94.563.813.019.3635.1
YOLOv10m94.363.532.2315.3112.6
YOLOv11m95.765.234.2620.0312.0
YOLOv12m95.465.434.0420.1015.8
YOLOv13L96.766.138.6727.6015.7
基线模型93.262.929.3120.0912.4
换主干模型95.063.715.9711.8035.6
本研究97.269.520.0715.3840.6
表 2  不同检测算法的性能对比实验结果
损失函数mAP@0.5/%R/%
GIoU93.288.5
InnerIoU93.189.5
MPDIoU94.988.1
Inner-MPDIoU95.791.0
表 3  损失函数的性能对比实验结果
图 12  不同损失函数训练过程中的性能对比
主干网络mAP@0.5/%Flops/109NP/106Rf/(帧· s?1)
GhostNetV292.213.6812.3839.9
MobileNetv293.712.319.3538.9
MobileNetv495.013.6411.0138.6
RepVit94.619.1513.3431.6
ShuffleNetV290.813.369.3538.7
EfficientNetV195.112.3313.8628.0
ConvNeXt94.716.6612.3036.4
UniRepLknet95.217.8012.9230.3
EfficientModel94.914.2213.1837.5
EfficientFormerV295.015.9711.8035.6
CG-EfficientFormerV295.316.3912.1740.0
表 4  轻量化主干网络的性能对比实验结果
注意力机制mAP@0.5/%Flops/109NP/106
基准模型93.229.3120.09
ECA92.929.8619.27
GAM93.532.4825.83
CBAM92.529.8719.53
MASAG94.832.3821.07
表 5  注意力机制的性能对比实验结果
模型mAP@0.5/%mAP@0.5:0.95/%Rf/(帧· s?1)
RT-DETR-R1842.826.021.6
RT-DETR-替换的主干43.526.735.3
RT-DETR-改进的主干45.027.534.0
本研究48.629.037.2
表 6  改进模型的泛化实验结果
图 13  自建数据集多维度统计分析
图 14  算法检测性能定量分析
图 15  实际检测结果和密集场景样本识别
图 16  不同光照场景识别和边缘/截断场景样本识别
图 17  不同模型的检测结果对比
1 张琴, 郭为安 深度学习小目标检测算法综述[J]. 计算机应用研究, 2025, 42 (10): 2893- 2904
ZHANG Qin, GUO Weian Survey on deep learning-based small object detection algorithms[J]. Application Research of Computers, 2025, 42 (10): 2893- 2904
2 刘颖, 刘红燕, 范九伦, 等 基于深度学习的小目标检测研究与应用综述[J]. 电子学报, 2020, 48 (3): 590- 601
LIU Ying, LIU Hongyan, FAN Jiulun, et al A survey of research and application of small object detection based on deep learning[J]. Acta Electronica Sinica, 2020, 48 (3): 590- 601
doi: 10.3969/j.issn.0372-2112.2020.03.024
3 REN S, HE K, GIRSHICK R, et al Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39 (6): 1137- 1149
doi: 10.1109/TPAMI.2016.2577031
4 LIN T Y, DOLLÁR P, GIRSHICK R, et al. Feature pyramid networks for object detection [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Honolulu: IEEE, 2017: 936–944.
5 WANG J, GAO J, ZHANG B A small object detection model in aerial images based on CPDD-YOLOv8[J]. Scientific Reports, 2025, 15: 770
doi: 10.1038/s41598-024-84938-4
6 WOO S, PARK J, LEE J Y, et al. CBAM: convolutional block attention module [C]// Computer Vision – ECCV 2018. Cham: Springer, 2018: 3–19.
7 TAN M, PANG R, LE Q V. EfficientDet: scalable and efficient object detection [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2020: 10778–10787.
8 ZHONG J, LIU Z, WANG H, et al A looseness detection method for railway catenary fasteners based on reinforcement learning refined localization[J]. IEEE Transactions on Instrumentation and Measurement, 2021, 70: 3518913
doi: 10.1109/tim.2021.3086913
9 WU X, DUAN J, YANG L, et al Intelligent cotter pins defect detection for electrified railway based on improved faster R-CNN and dilated convolution[J]. Computers in Industry, 2024, 162: 104146
doi: 10.1016/j.compind.2024.104146
10 ZHAO Y, LV W, XU S, et al. DETRs beat YOLOs on real-time object detection [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Seattle: IEEE, 2024: 16965–16974.
11 LI Y, HU J, WEN Y, et al. Rethinking vision transformers for MobileNet size and speed [C]// Proceedings of the IEEE/CVF International Conference on Computer Vision. Paris: IEEE, 2024: 16843–16854.
12 WU T, TANG S, ZHANG R, et al CGNet: a light-weight context guided network for semantic segmentation[J]. IEEE Transactions on Image Processing, 2021, 30: 1169- 1179
doi: 10.1109/TIP.2020.3042065
13 KOLAHI S G, CHAHARSOOGHI S K, KHATIBI T, et al. MSA2Net: multi-scale adaptive attention-guided network for medical image segmentation [C]// Proceedings of the 35th British Machine Vision Conference. Glasgow: [s.n.], 2024: 1–13.
14 LIU W, LU H, FU H, et al. Learning to upsample by learning to sample [C]// Proceedings of the IEEE/CVF International Conference on Computer Vision. Paris: IEEE, 2024: 6004–6014.
15 XU C, WANG J, YANG W, et al Detecting tiny objects in aerial images: a normalized Wasserstein distance and a new benchmark[J]. ISPRS Journal of Photogrammetry and Remote Sensing, 2022, 190: 79- 93
doi: 10.1016/j.isprsjprs.2022.06.002
16 REZATOFIGHI H, TSOI N, GWAK J, et al. Generalized intersection over union: a metric and a loss for bounding box regression [C]// Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Long Beach: IEEE, 2020: 658–666.
17 ZHANG H, XU C, ZHANG S. Inner-IoU: more effective intersection over union loss with auxiliary bounding box [EB/OL]. (2023–11–06)[2025–11–08]. https://arxiv.org/pdf/2311.02877.
18 MA S, XU Y. MPDIoU: a loss for efficient and accurate bounding box regression [EB/OL]. (2023–07–14)[2025–11–08]. https://arxiv.org/pdf/2307.07662.
[1] 李珺,丁彬彬,史维娟,杨琳. 基于多尺度特征聚合的航拍图像检测算法[J]. 浙江大学学报(工学版), 2026, 60(9): 1890-1900.
[2] 董绍江,肖涛,吕振鸣,夏浩然,罗家元,孙世政,张霞,刘超. 基于改进RT-DETR的水下色偏环境中小型生物检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1404-1415.
[3] 肖剑,杨小苑,何昕泽,陈林,胡欣. 基于全局信息感知的轻量级螺纹钢表面缺陷检测算法[J]. 浙江大学学报(工学版), 2026, 60(7): 1438-1451.
[4] 吴佰靖,闫光辉,马龙,程文鑫,黄亚宁. 边缘感知和跨尺度特征增强的小目标水漂垃圾检测[J]. 浙江大学学报(工学版), 2026, 60(7): 1452-1463.
[5] 武晓春,郭宁. 基于HMARU-net的隧道渗漏水轻量化检测方法[J]. 浙江大学学报(工学版), 2026, 60(3): 468-477.
[6] 李彬彬,张超,覃涛,陈昌盛,刘兴艳,杨靖. 面向光伏电站建设的移动端人体跌倒检测方法[J]. 浙江大学学报(工学版), 2026, 60(3): 546-555.
[7] 雒伟群,陆敬蔚,吴佳缔,梁钰迎,申传鹏,朱睿. 藏区高原典型环境地形目标的轻量化检测模型[J]. 浙江大学学报(工学版), 2026, 60(3): 594-603.
[8] 刘慧,王防修,王意,黄淄博,苏晨. 轻量级改进RT-DETR的葡萄叶片病害检测算法[J]. 浙江大学学报(工学版), 2026, 60(3): 604-613.
[9] 方芳,严军,郭红想,王勇. 基于时空注意力机制的轻量级脑纹识别算法[J]. 浙江大学学报(工学版), 2026, 60(3): 633-642.
[10] 董超群,汪战,廖平,谢帅,荣玉杰,周靖淞. 轻量化YOLOv5s-OCG的轨枕裂纹检测算法[J]. 浙江大学学报(工学版), 2025, 59(9): 1838-1845.
[11] 翟亚红,陈雅玲,徐龙艳,龚玉. 改进YOLOv8s的轻量级无人机航拍小目标检测算法[J]. 浙江大学学报(工学版), 2025, 59(8): 1708-1717.
[12] 梁耕良,韩曙光. 基于改进RT-DETR的牛仔面料疵点检测算法[J]. 浙江大学学报(工学版), 2025, 59(6): 1169-1178.
[13] 邢国华,陆勇健,苗鹏勇,陈思锦. 基于改进遗传算法的临时转播塔结构优化设计方法[J]. 浙江大学学报(工学版), 2025, 59(3): 469-479.
[14] 梁礼明,龙鹏威,金家新,李仁杰,曾璐. 基于改进YOLOv8s的钢材表面缺陷检测算法[J]. 浙江大学学报(工学版), 2025, 59(3): 512-522.
[15] 刘登峰,陈世海,郭文静,柴志雷. 基于轻量残差网络的高效半色调算法[J]. 浙江大学学报(工学版), 2025, 59(1): 62-69.