LKAM-RCatenaryDet: enhanced railway catenary foreign object detection via large kernel fusion and global attention mechanism
摘要
The railway catenary is central to railway electrification. Because of its exposed location and long spans, the catenary system is susceptible to the intrusion of foreign objects such as plastic bags, bird nests, and kites, which reduce the current collection performance of electric locomotives and increase the safety risks associated with railway operations. Due to the small target size and complex background environment, detecting foreign objects on the catenary poses significant challenges. For catenary foreign object detection which is challenged by small targets and complex backgrounds, this paper presents LKAM-RCatenaryDet, which incorporates a reparameterization-optimized large convolution kernel and a global attention mechanism. To enhance the capability of small-target detection, the LRFSRBackbone network is designed. With the goal of strengthening complex background detection, the Global Attention Mechanism is integrated into the small-target detection head. Furthermore, the Wise-IoU model is adopted as the loss function, which addresses anchor box overlap and refines detection accuracy. To verify the effectiveness of the proposed LKAM-RCatenaryDet, experiments were conducted on the self-constructed railway catenary foreign object dataset, with results showing that LKAM-RCatenaryDet surpasses YOLO11s which is currently recognized as the detection model in this field in Precision (+ 3.2%), Recall (+ 4.2%), mAP@50 (+ 4.7%), and mAP@95 (+ 2.6%). Dataset and code location: https://github.com/DJTUResearches/LKAM.