Lightweight Detection of Train Underframe Bolts Based on SFCA-YOLOv8s

Abstract

Improving the accuracy and detection speed of bolt recognition under the complex background of the train underframe is crucial for the safety of train operation. To achieve efficient detection, a lightweight detection method based on SFCA-YOLOv8s is proposed. The underframe bolt images are captured by a self-designed track-based inspection robot, and a dataset is constructed by mixing simulated platform images with real train underframe bolt images. By combining the C2f module with ScConv lightweight convolution and replacing the Bottleneck structure with the Faster_Block structure, the SFC2f module is designed for feature extraction to improve detection accuracy and speed. It is compared with FasterNet, GhostNet, and MobileNetV3. Additionally, the CA attention mechanism is introduced, and MPDIoU is used as the loss function of YOLOv8s. LAMP scores are used to rank the model weight parameters, and unimportant weight parameters are pruned to achieve model compression. The compressed SFCA-YOLOv8s model is compared with models such as YOLOv5s, YOLOv7, and YOLOX-s in comparative experiments. The results indicate that the final model achieves an average detection accuracy of 93.3% on the mixed dataset, with a detection speed of 261 FPS. Compared with other classical deep learning models, the improved model demonstrates superior performance in detection effectiveness, robustness, and generalization. Even in the absence of sufficient real underframe bolt images, the algorithm enables the trained network to better adapt to real environments, improving bolt recognition accuracy and detection speed, thus providing technical references and theoretical support for subsequent related research.