Real-time detection of railway cracks based on improved YOLOX-Nano

Chong Du; Xiaowei Zao; Xiaoliang Wu

doi:10.1117/12.2667626

28 March 2023 Real-time detection of railway cracks based on improved YOLOX-Nano

Chong Du, Xiaowei Zao, Xiaoliang Wu

Proceedings Volume 12566, Fifth International Conference on Computer Information Science and Artificial Intelligence (CISAI 2022); 125662T (2023) https://doi.org/10.1117/12.2667626
Event: Fifth International Conference on Computer Information Science and Artificial Intelligence (CISAI 2022), 2022, Chongqing, China

Abstract

Cracks in the rails will lead to great safety hazards in railway transportation. Aiming at the problems of low detection accuracy and inconspicuous part of cracks in crack detection, an improved model based on YOLOX-Nano is proposed. The SA-NET lightweight combined attention mechanism is added to the model to generate a feature map with channel attention and spatial attention, which strengthens the model's attention to target features and location information. Secondly, use Alpha-CIoU Loss to replace IoU Loss to increase the accuracy of the model's prediction box. The comparison experiment was carried out on the self-built data set, and the mAP of the improved YOLOX-Nano model reached 77.58%, the detection speed reached 42.2FPS, and the calculation amount and parameter amount of the model were only 0.508G and 3.5MB respectively, and the overall performance was better than other models.

Citation Download Citation

Chong Du, Xiaowei Zao, and Xiaoliang Wu "Real-time detection of railway cracks based on improved YOLOX-Nano", Proc. SPIE 12566, Fifth International Conference on Computer Information Science and Artificial Intelligence (CISAI 2022), 125662T (28 March 2023); https://doi.org/10.1117/12.2667626

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available