High-Speed Rail Tunnel Panoramic Inspection Image Recognition Technology Based on Improved YOLOv5

Abstract

In order to meet the fast and accurate automatic detection requirements of equipment maintenance in railway tunnels in the era of high-speed railways, as well as adapting to the high dynamic, low-illumination imaging environment formed by strong light at the tunnel exit, we propose an automatic inspection solution based on panoramic imaging and object recognition with deep learning. We installed a hyperboloid catadioptric panoramic imaging system on an inspection vehicle to obtain a large field of view as well as to shield the high dynamic phenomena at the tunnel exit, and proposed a YOLOv5-CCFE object detection model based on railway equipment recognition. The experimental results show that the [email protected] value of the YOLOv5-CCFE model reaches 98.6%, and [email protected]:0.95 reaches 68.9%. The FPS value is 158, which can meet the automatic inspection requirements of railway tunnel equipment along the line and has high practical application value.

Keywords

• railway tunnel
• catadioptric panoramic imaging system
• YOLOv5
• CARAFE
• CBAM
• loss function