Projects per year
Abstract
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 predictions of categories and bounding boxes are generated via two feed forward networks 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.
| Original language | English |
|---|---|
| Pages (from-to) | 2379-2388 |
| Journal | IEEE Sensors Journal |
| Volume | 22 |
| Issue number | 3 |
| Online published | 3 Dec 2021 |
| DOIs | |
| Publication status | Published - 1 Feb 2022 |
Research Keywords
- Attention mechanism
- data augmentation
- deep learning
- Fasteners
- Feature extraction
- rail component detection
- Rail transportation
- Rails
- railway inspection
- Sensors
- Task analysis
- Training
Fingerprint
Dive into the research topics of 'Automatic Rail Component Detection Based on AttnConv-net'. Together they form a unique fingerprint.Projects
- 3 Finished
-
RIF-ExtU-Lead: Enhancing Safety, Punctuality and Ride Comfort of Railway Transportation: From Local Metro System to Global High-speed Rail Network
NI, Y. Q. (Main Project Coordinator [External]) & LO, S. M. (Principal Investigator / Project Coordinator)
1/06/19 → 26/10/23
Project: Research
-
GRF: A Collaborative Data-driven Methodology for Improving Wind Farm Operations and Maintenance
ZHANG, Z. (Principal Investigator / Project Coordinator)
1/01/19 → 7/06/23
Project: Research
-
TBRS: Safety, Reliability, and Disruption Management of High Speed Rail and Metro Systems
XIE, M. (Principal Investigator / Project Coordinator), BENSOUSSAN, A. (Co-Principal Investigator), LO, S. M. (Co-Principal Investigator), SHOU, B. (Co-Principal Investigator), SINGPURWALLA, N. D. (Co-Principal Investigator), TSE, W. T. P. (Co-Principal Investigator), TSUI, K. L. (Co-Principal Investigator), YU, Y. (Co-Principal Investigator), YUEN, K. K. R. (Co-Principal Investigator), CHAN, A. B. (Co-Investigator), CHAN, N.-H. (Co-Investigator), CHIN, K. S. (Co-Investigator), CHOW, H. A. (Co-Investigator), Chow, W. K. (Co-Investigator), EDESESS, M. (Co-Investigator), GOLDSMAN, D. M. (Co-Investigator), Huang, J. (Co-Investigator), LEE, W. M. (Co-Investigator), LI, L. (Co-Investigator), LI, C. L. (Co-Investigator), LING, M. H. A. (Co-Investigator), LIU, S. (Co-Investigator), MURAKAMI, J. (Co-Investigator), NG, S. Y. S. (Co-Investigator), NI, M. C. (Co-Investigator), TAN, M.H.-Y. (Co-Investigator), Wang, W. (Co-Investigator), Wang, J. (Co-Investigator), WONG, C. K. (Co-Investigator), WONG, S. Y. Z. (Co-Investigator), WONG, S. C. (Co-Investigator), Xu, Z. (Co-Investigator), ZHANG, Z. (Co-Investigator), Zhang, D. (Co-Investigator), ZHAO, J. L. (Co-Investigator) & Zhou, Q. (Co-Investigator)
1/01/16 → 31/12/21
Project: Research