Improved YOLOv8 for high-precision detection of rail surface defects on heavy-haul railways

The complex infrastructure and harsh conditions of heavy-haul railways result in frequent and rapidly deteriorating rail surface defects. Accurate detection of these defects is essential. To solve the problem of low detection precision caused by complex background interference, significant variation in defect scales, and similar features between different types of defects, a high-precision rail surface defect detection method for heavy-haul railways based on an improved YOLOv8 is proposed. First, the original grayscale images are preprocessed to reduce background noise interference. Then, the designed Scale Variation Adaptation Module (SVAM) is introduced to mitigate the impact of significant scale variations in the target defects. Additionally, a Bidirectional Feature Pyramid Network (Bi-FPN) is incorporated to enhance feature fusion effectiveness. Furthermore, a small target detection head is introduced to improve the detection performance of small-scale defects. Lastly, network performance is optimized by replacing the original loss function with Wise-IoU (WIoU). Experimental results demonstrate that the improved model achieves a mAP50 value of 0.975, representing a 4.13% improvement in Precision and a 7.75% increase in Recall compared to the baseline model. The improved model effectively detects typical defects such as Spalling, Shelling, and Corrugation, providing valuable technical support for field maintenance personnel.