7 August 2024 DeepLab-Rail: semantic segmentation network for railway scenes based on encoder-decoder structure
Qingsong Zeng, Linxuan Zhang, Yuan Wang, Xiaolong Luo, Yannan Chen
Author Affiliations +
Abstract

Understanding the perimeter objects and environment changes in railway scenes is crucial for ensuring the safety of train operation. Semantic segmentation is the basis of intelligent perception and scene understanding. Railway scene categories are complex and effective features are challenging to extract. This work proposes a semantic segmentation network DeepLab-Rail based on classic yet effective encoder-decoder structure. It contains a lightweight feature extraction backbone embedded with channel attention (CA) mechanism to keep computational complexity low. To enrich the receptive fields of convolutional modules, we design a parallel and cascade convolution module called compound-atrous spatial pyramid pooling and a combination of dilated convolution ratio is selected through experiments to obtain multi-scale features. To fully use the shallow features and the high-level features, efficient CA mechanism is introduced and also the mixed loss function is designed for the problem of unbalanced label categories of the dataset. Finally, the experimental results on the RailSem19 railway dataset show that the mean intersection over union reaches 65.52% and the PA reaches 88.48%. The segmentation performance of railway confusing facilities, such as signal lights and catenary pillars, has been significantly improved and surpasses other advanced methods to our best knowledge.

© 2024 SPIE and IS&T
Qingsong Zeng, Linxuan Zhang, Yuan Wang, Xiaolong Luo, and Yannan Chen "DeepLab-Rail: semantic segmentation network for railway scenes based on encoder-decoder structure," Journal of Electronic Imaging 33(4), 043038 (7 August 2024). https://doi.org/10.1117/1.JEI.33.4.043038
Received: 16 May 2024; Accepted: 16 July 2024; Published: 7 August 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Semantics

Image segmentation

Feature extraction

Convolution

Education and training

Data modeling

Detection and tracking algorithms

Back to Top