Paper
15 November 2022 Damage detection of train wheelset tread based on improved SSD
Author Affiliations +
Proceedings Volume 12448, 5th Optics Young Scientist Summit (OYSS 2022); 124480G (2022) https://doi.org/10.1117/12.2636170
Event: 5th Optics Young Scientist Summit (OYSS 2022), 2022, Fuzhou, China
Abstract
The train wheelset is a crucial part of railway vehicles, and its damage may lead to serious safety accidents. Therefore, it is imperative to detect tread damage timely and accurately. With the rapid development of deep learning, the image detection method based on a convolutional neural network (CNN) has played an important role. Single Shot MultiBox Detector (SSD) is one of the fastest algorithms in the target detection field. The algorithm has achieved excellent results in target detection, but there is a low recognition rate for small targets. Therefore, we propose an improved SSD target detection algorithm. The Original SSD algorithm is ineffective in detecting small targets with pits and cracks, so conv3-3 is selected to join the detection. We optimize convolution kernel parameters; the convolution layer contains more small target details. Compared with the original SSD, the Mean Average Precision (MAP) of tread defect is improved by 4.38%, and the MAP of small target detection is enhanced by 7.24%. This algorithm has a better performance in detection accuracy.
© (2022) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Rui Wang, Zhi-Feng Zhang, Ben Yang, Hai-Qi Xi, Peng Yang, Rui-Liang Zhang, Li-Jie Geng, Yu-Sheng Zhai, Kun Yang, and Dong-Lin Wang "Damage detection of train wheelset tread based on improved SSD", Proc. SPIE 12448, 5th Optics Young Scientist Summit (OYSS 2022), 124480G (15 November 2022); https://doi.org/10.1117/12.2636170
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Target detection

Damage detection

Convolutional neural networks

Image enhancement

Sensors

Back to Top