A lightweight transformer-based framework for real-time foreign object detection in complex railway environments
摘要
The safe operation of railway systems critically depends on accurate and real-time detection of foreign objects intruding into track areas. To address the limitations of traditional manual monitoring methods, which often suffer from delayed response and misjudgment in complex environments, this study proposes CDH-DETR, a lightweight Transformer-based real-time detection model for railway foreign objects. Building upon the RT-DETR model, our model introduces three key structural innovations to achieve an optimal balance between accuracy, speed, and model complexity. The architecture incorporates a ContextGuided Block module to reconstruct the backbone network, substantially reducing parameter count and computational complexity while maintaining feature representation capability. Furthermore, the model employs the Dynamic upsampling (DySample) to replace traditional operations, enhancing detail perception and restoring critical features in complex scenes without increasing computational burden. Additionally, a dedicated High-level Screening-feature Fusion Pyramid Networks (HSFPN) strengthens multi-scale feature fusion and improves detection robustness for targets with significant scale variations. Experimental results on a constructed railway intrusion dataset demonstrate that CDH-DETR achieves 76.7%