학술논문

Key Point Estimate Network for Rail-Track Detection
Document Type
Periodical
Source
IEEE Transactions on Intelligent Transportation Systems IEEE Trans. Intell. Transport. Syst. Intelligent Transportation Systems, IEEE Transactions on. 25(5):4077-4088 May, 2024
Subject
Transportation
Aerospace
Communication, Networking and Broadcast Technologies
Computing and Processing
Robotics and Control Systems
Signal Processing and Analysis
Rails
Transportation
Roads
Rail transportation
Lighting
Accidents
Task analysis
Rail-track detection
pseudo-attention
dislocation assignment
rail-track generalized IoU
Language
ISSN
1524-9050
1558-0016
Abstract
Rail-track detection is a crucial function for an active obstacle avoidance system in trains. However, existing methods face challenges in effectively detecting rail-tracks, particularly in turnout scenarios. This study introduces a novel rail-track detection approach using a key-point estimate network. The network treats the rail-track as a pair and constructs a dedicated model for detection. Additionally, a pseudo-attention mechanism leverages the detection output from previous stages, enabling the network to focus on the rail-track region. Also, a dislocation assignment mechanism is proposed to address label assignment confusion at turnouts. Moreover, a rail-track generalized IoU is also introduced, treating the rail-track as a pair and adds a correction term to enhance detection performance. Experimental results demonstrate that the proposed method achieves a remarkable mF1 score of 69.42%, establishing it as the state-of-the-art (SOTA) in this field. Furthermore, the effectiveness of the proposed method has been validated and applied in real-world testing on the Hong Kong Metro Tsuen Wan Line.