An Efficient Foreign Object Recognition Model in Rail Transit Based on Real-Time Railway Region Extraction and Object Detection

Zhi-Cheng Feng,Jie Yang,Fan Li,Zhi-Chao Chen,Zhuang Kang,Li-Min Jia
DOI: https://doi.org/10.1007/s42835-024-01805-y
2024-02-09
Abstract:The existing foreign object recognition methods have some limitations, such as imprecise extraction of railway regions and limited real-time performance. This study proposes a efficient railway region extraction model, called Railseg. Railseg is based on lightweight principles and consist of an encoder, an adaptive decoder and a segmentation head. The encoder is constructed by sandglass unit, effectively utilizing the convolution operator to encode features. The adaptive decoder can fuse context and spatial information based on the index-query mechanism. The segmentation head introduces ASPP (Atrous Spatial Pyramid Pooling) to extract multi scale features. As a result, the number of parameters and FLOPs (Floating Point Operations) for Railseg are 1.54 M and 1.56 G, respectively. Considering real-time and segmentation performance, Railseg is more advantageous than some mainstream models, reaching 15 FPS (Frames Per Second) and 91.6% of mIoU (mean Intersection over Union) on the Jetson TX2. Then, based on the object detection model Yolox to capture the location information of persons, which achieves 36 FPS and AP (Average Precision) of 85.3% on the Jetson TX2. Finally, this study can recognize foreign object through the correlation between person and railway, which is positive for traffic safety.
engineering, electrical & electronic
What problem does this paper attempt to address?