Please wait a minute...
Journal of ZheJiang University (Engineering Science)  2026, Vol. 60 Issue (11): 2418-2430    DOI: 10.3785/j.issn.1008-973X.2026.11.011
    
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
Download: HTML     PDF(3556KB) HTML
Export: BibTeX | EndNote (RIS)      

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     
Received: 14 September 2025      Published: 16 October 2026
CLC:  TP 393  
Fund:  国家自然科学基金资助项目(61661027);中国国家铁路集团有限公司基金资助项目(N2022G012).
Cite this article:

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.

URL:

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


基于改进RT-DETR的接触网吊弦螺栓小目标故障检测算法

铁路接触网吊弦螺栓故障检测中存在目标尺寸小、特征提取困难的问题,为此提出基于改进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,  小目标故障检测,  轻量化,  上下文引导模块 
Fig.1 RT-DETR model
Fig.2 Improved RT-DETR model
Fig.3 Schematic of context-guided block
Fig.4 CG-EfficientFormerV2 network architecture
Fig.5 DySample dynamic upsampling process
Fig.6 Sampling generator in DySample
Fig.7 Spatial-channel guided multi-scale attention mechanism
Fig.8 Schematic of Inner-MPDIou joint loss function
Fig.9 Dataset preprocessing
Fig.10 Adaptive histogram equalization algorithm
Fig.11 Normal and defective samples of catenary dropper bolts
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
Tab.1 Ablation experimental results of improved modules
算法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
Tab.2 Performance comparison results of different detection algorithms
损失函数mAP@0.5/%R/%
GIoU93.288.5
InnerIoU93.189.5
MPDIoU94.988.1
Inner-MPDIoU95.791.0
Tab.3 Performance comparison results of loss functions
Fig.12 Performance comparison of different loss functions during training
主干网络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
Tab.4 Performance comparison results of lightweight backbone networks
注意力机制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
Tab.5 Performance comparison results of attention mechanisms
模型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
Tab.6 Generalization experiment results of improved model
Fig.13 Multi-dimensional statistical analysis of custom-built dataset
Fig.14 Quantitative analysis of algorithm’s detection performance
Fig.15 Actual detection results and dense scene sample recognition
Fig.16 Sample recognition under different lighting conditions and edge/truncated scenarios
Fig.17 Comparison of detection results across different models
[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] Jun LI,Binbin DING,Weijuan SHI,Lin YANG. Aerial image detection algorithm based on multiscale feature aggregation[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1890-1900.
[2] Hongwei LI,Jianyang DING,Zhilei CHAI. Data-efficient lightweight model for real-time semantic segmentation[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1934-1941.
[3] Peng CUI,Chuang GAO,Qinghan MENG. Lightweight frequency-domain super-resolution network based on amplitude-phase collaboration in complex domain[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(9): 1962-1971.
[4] Liming LIANG,Chengbin WANG,Yi ZHONG,Linjun CHEN,Jian WU. Retinal vessel segmentation based on lightweight high-frequency Transformer and feature complementary fusion[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(7): 1392-1403.
[5] Shaojiang DONG,Tao XIAO,Zhenming LV,Haoran XIA,Jiayuan LUO,Shizheng SUN,Xia ZHANG,Chao LIU. Small organism detection in underwater color-cast environments based on improved RT-DETR[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(7): 1404-1415.
[6] Jian XIAO,Xiaoyuan YANG,Xinze HE,Lin CHEN,Xin HU. Lightweight rebar surface defect detection algorithm based on global information perception[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(7): 1438-1451.
[7] Baijing WU,Guanghui YAN,Long MA,Wenxin CHENG,Yaning HUANG. Small-target water-floating garbage detection based on edge perception and cross-scale feature enhancement[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(7): 1452-1463.
[8] Xiaochun WU,Ning GUO. Lightweight detection method of water leakage in tunnel based on HMARU-net[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(3): 468-477.
[9] Binbin LI,Chao ZHANG,Tao QIN,Changsheng CHEN,Xingyan LIU,Jing YANG. Mobile-based human fall detection method for photovoltaic power plant construction[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(3): 546-555.
[10] Weiqun LUO,Jingwei LU,Jiadi WU,Yuying LIANG,Chuanpeng SHEN,Rui ZHU. Lightweight detection model for typical environmental terrain target in Tibetan Plateau[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(3): 594-603.
[11] Hui LIU,Fangxiu WANG,Yi WANG,Zibo HUANG,Chen SU. Lightweight improved RT-DETR algorithm for grape leaf disease detection[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(3): 604-613.
[12] Fang FANG,Jun YAN,Hongxiang GUO,Yong WANG. Lightweight brainprint recognition algorithm based on spatio-temporal attention mechanism[J]. Journal of ZheJiang University (Engineering Science), 2026, 60(3): 633-642.
[13] Chaoqun DONG,Zhan WANG,Ping LIAO,Shuai XIE,Yujie RONG,Jingsong ZHOU. Lightweight YOLOv5s-OCG rail sleeper crack detection algorithm[J]. Journal of ZheJiang University (Engineering Science), 2025, 59(9): 1838-1845.
[14] Yahong ZHAI,Yaling CHEN,Longyan XU,Yu GONG. Improved YOLOv8s lightweight small target detection algorithm of UAV aerial image[J]. Journal of ZheJiang University (Engineering Science), 2025, 59(8): 1708-1717.
[15] Gengliang LIANG,Shuguang HAN. Denim fabric defect detection algorithm based on improved RT-DETR[J]. Journal of ZheJiang University (Engineering Science), 2025, 59(6): 1169-1178.