Enhanced Detection of Subway Insulator Defects Based on Improved YOLOv5

Abstract

Insulators, pivotal to the integrity of railway catenaries, demand impeccable functioning to prevent system failures. Their consistent assessment is vital for railway safety. Current insulator evaluations in subways predominantly involve human intervention, a method fraught with inefficiencies, inaccuracies, and oversights, exacerbated by the complex backdrop of subway tunnels and minuscule defect dimensions. This study introduces an enhanced algorithm, anchored in the YOLOv5 framework, to refine insulator defect identification. Challenges in defect detection include limited, imbalanced data samples and adaptability. Addressing this, an accurate catenary model mirrors the subway line’s architecture, facilitating the creation of synthetic instances of both intact and impaired insulators. An atomization technique augments the dataset volume, fortifying the algorithm’s resilience in reduced visibility conditions, such as fog. Tackling sample equilibrium, the study introduces an equilibrium loss function, assigning disparate weights to various sample categories during training, thereby sharpening the algorithm’s focus on positive instances, particularly those that are challenging to discern, and rectifying the disproportion in sample categories. Incorporating lightweight structures like GhostNet and the Efficient Channel Attention Network (ECA-Net) channel attention scheme not only diminishes the network’s computational demands, thereby elevating the detection capabilities, but also minimizes superfluous data processing, enhancing the accuracy in identifying smaller targets. Empirical analyses indicate substantial model optimization: a size reduction to 60 pp of its original (from 15 MB to 9 MB), a near 1.4 pp increase in mean average precision (mAP) to 96.57%, and a tripling of the detection speed (from 30 to 90 FPS). Real-world image assessments further reveal a mAP improvement of approximately 2.5 pp (reaching 98.43%), confirming the model’s suitability for real-time applications.