机电工程技术2026，Vol.55Issue(8)：38-43,6.DOI:10.3969/j.issn.1009-9492.2025.00083
基于Transformer与MobileNetv3融合的铁路钢轨表面伤损检测算法
Rail Track Surface Defect Detection Algorithm Integrating MobileNetv3 and Transformer
摘要
Abstract
Surface defects on railway rails,such as micro-cracks and corrosion,are among the critical factors affecting the safety of train operations.To address the efficiency and accuracy bottlenecks in detecting these minute defects on rail surfaces,this study proposes an intelligent recognition method for rail surface damage based on the collaborative optimization of MobileNetv3 and Transformer.By injecting spatially coordinate-aware features into the channel attention mechanism,an improved CBAM-Bneck module is integrated into the MobileNetv3 architecture,effectively enhancing the network's ability to discern and generalize rail defect features.A novel MobileNetV3-CBTr composite backbone network is constructed,comprising MobileNetv3 base modules,enhanced CBAM-Bneck units,and a lightweight Transformer encoding layer.This design ensures robust feature representation while significantly reducing the model's parameter count.A BiFPN-Lite is introduced to efficiently fuse multi-scale defect features without increasing computational load.The optimized YOLO detection head is then employed for precise localization and classification of rail damages.Experimental results on a self-built rail dataset demonstrate that the proposed algorithm achieves a mAP of 91.8%and a processing speed of 19.5 f/s,representing a 3.5%improvement over YOLOv5.This indicates that the method can effectively accomplish high-precision detection of railway rail surface damages.关键词
伤损检测/MobileNetv3/CBAM注意力模块/Transformer模块Key words
damage detection/MobileNetv3/CBMA attention module/Transformer module分类
信息技术与安全科学引用本文复制引用
宁善平,符秀芬,江铭臻,武文星,江远平..基于Transformer与MobileNetv3融合的铁路钢轨表面伤损检测算法[J].机电工程技术,2026,55(8):38-43,6.基金项目
2024年广东省科技创新战略专项资金(大学生科技创新培育)(pdjh2024b573) （大学生科技创新培育）
广东省普通高校特色创新类项目(2024KTSCX381) （2024KTSCX381）
广东交通职业技术学院大学生科技创新项目(GDCP-ZX-2024-016-N2,GDCP-ZX-2023-031-N6) （GDCP-ZX-2024-016-N2,GDCP-ZX-2023-031-N6）
