Enhancing rail safety through real-time defect detection: A novel lightweight network approach

Document Type

Journal Article

Publication Date

2024

Subject Area

mode - rail, infrastructure - maintainance, infrastructure - track, technology - intelligent transport systems, planning - safety/accidents, planning - methods

Keywords

Rail, safety, defect detection

Abstract

The rapid detection of internal rail defects is critical to maintaining railway safety, but this task faces a significant challenge due to the limited computational resources of onboard detection systems. This paper presents YOLOv8n-LiteCBAM, an advanced network designed to enhance the efficiency of rail defect detection. The network designs a lightweight DepthStackNet backbone to replace the existing CSPDarkNet backbone. Further optimization is achieved through model pruning techniques and the incorporation of a novel Bidirectional Convolutional Block Attention Module (BiCBAM). Additionally, inference acceleration is realized via ONNX Runtime. Experimental results on the rail defect dataset demonstrate that our model achieves 92.9% mAP with inference speeds of 136.79 FPS on the GPU and 38.36 FPS on the CPU. The model’s inference speed outperforms that of other lightweight models and ensures that it meets the real-time detection requirements of Rail Flaw Detection (RFD) vehicles traveling at 80 km/h. Consequently, the YOLOv8n-LiteCBAM network is with some potential for industrial application in the expedited detection of internal rail defects.

Rights

Permission to publish the abstract has been given by Elsevier, copyright remains with them.

Comments

Accident Analysis and Prevention Home Page:

http://www.sciencedirect.com/science/journal/00014575

Link to Full Text

Share

COinS