返回
Automatic Rail Component Detection Based on AttnConv-Net
DOI:10.1109/JSEN.2021.3132460.png)
摘要
En 中文
The automatic detection of major rail components using railway images is beneficial to ensure the rail transport safety. In this paper, we propose an attention-powered deep convolutional network (AttnConv-net) to detect multiple rail components including the rail, clips, and bolts. The proposed method consists of a deep convolutional neural network (DCNN) as the backbone, cascading attention blocks (CAB), and two feed forward networks (FFN). Two types of positional embedding are applied to enrich information in latent features extracted from the backbone. Based on processed latent features, the CAB aims to learn the local context of rail components including their categories and component boundaries. Final categories and bounding boxes are generated via two FFN implemented in parallel. To enhance the detection of small components, various data augmentation methods are employed in training process. The effectiveness of the proposed AttnConv-net is validated with one real dataset and another synthesized dataset. Compared with classic convolutional neural network based methods, our proposed method simplifies the detection pipeline by eliminating the need of prior- and post-processing, which offers a new speed-quality solution to enable faster and more accurate image-based rail component detections.
Keyword:
Rails
Fasteners
Sensors
Rail transportation
Feature extraction
Training
Task analysis
Attention mechanism
data augmentation
deep learning
rail component detection
railway inspection
期刊
IF:
4.5
论文数:
2.2W
被引数:
7.3W
机构
引用论文
Acoustic emission detection of rail defect based on wavelet transform and Shannon entropy基于小波变换和Shannon熵的钢轨缺陷声发射检测
Obstacle Detection on Railway Tracks Using Vibration Sensors and Signal Filtering Using Bayesian Analysis使用振动传感器检测铁路轨道上的障碍物,并使用贝叶斯分析进行信号滤波
IEEE SENSORS JOURNAL
IF4.5

