Bulletin of Surveying and Mapping ›› 2024, Vol. 0 ›› Issue (11): 108-114.doi: 10.13474/j.cnki.11-2246.2024.1119

Previous Articles    

Target detection method of railway catenary components in UAV images based on improved YOLOv7

SONG Zongying1, WANG Xingzhong1, ZENG Shan2, ZHANG Zhengjun3, YIN Taijun3, LIU Hongli3   

  1. 1. China Shenhua Energy Co., Ltd., Beijing 100011, China;
    2. Institute of Geographic Sciences and Natural Research, CAS, Beijing 100101, China;
    3. Digitwinology International Co., Ltd., Qinhuangdao 066000, China
  • Received:2024-07-01 Published:2024-12-05

Abstract: The catenary system, as an essential component of electrified railways, provides energy to trains and ensures their normal operation. Damage to catenary system components poses a threat to train safety, making it crucial to regularly inspect the condition of these components. In recent years, UAVs have been widely used in monitoring the condition of critical catenary components. However, due to the complex and variable backgrounds, significant scale changes, and the presence of many small targets in the catenary images captured by UAVs, existing detection algorithms frequently suffer from false detections and missed detections of catenary components. To address this issue, this paper proposes a catenary component target detection method based on an improved YOLOv7 algorithm. By introducing an enhanced receptive field module, the network's feature extraction capability is strengthened, leading to more discriminative target feature representations. Additionally, an improved coordinate attention mechanism is incorporated during the fusion of adjacent scale feature maps to highlight the target features of catenary components and suppress redundant background information. The bounding box loss function is optimized using the Wasserstein distance, effectively improving detection accuracy. Experiments on the catenary component dataset show that the improved YOLOv7 algorithm can accurately detect various catenary components in drone-captured images, achieving a mean average precision of 97.27%, which is 3.83% higher than before the improvement. The proposed algorithm enhances the high-precision and rapid detection capabilities of drones for critical catenary components, providing technical support for achieving more intelligent drone inspections.

Key words: object detection, YOLOv7, contact network component recognition, feature extraction, attention mechanism, loss function

CLC Number: