A fault detection method for transmission line components based on synthetic dataset and improved YOLOv5

摘要

To address insufficient fault images for transmission line components (TLCs) and low accuracy in fault detection using deep learning techniques, we propose a fault detection method for TLCs based on synthetic datasets and improved YOLOv5. First, we introduce a synthetic image approach that uses prior information from the inspection process of a developed flying-walking power transmission line inspection robot (FPTLIR) to generate a synthetic dataset of fault components (SDFC). Second, we propose an improved YOLOv5 network called CSH-YOLOv5 to improve the accuracy of fault detection. The CSH-YOLOv5 network incorporates the convolutional block attention module (CBAM) and the latest SimCSPSPPF module to improve the detection accuracy of the network. In addition, a statistical analysis of small objects in the SDFC is performed and the neck and head of the YOLOv5 network are optimized accordingly to detect small objects. Finally, to address the lack of fault images in the unmanned aerial vehicle (UAV) inspection dataset, we implement a two-stage transfer learning strategy using the SDFC for training. We then experimentally evaluate the performance of the CSH-YOLOv5 network on a real test dataset. The results show that the CSH-YOLOv5 network achieves a mAP@[0.5] of 98.0% and a mAP@[0.5: 0.95] of 64.6% for fault detection, representing an improvement of 7.6% and 8.1%, respectively, over the YOLOv5 network. Comparative analysis indicates that the CSH-YOLOv5 outperforms other popular object detection networks, including Faster-RCNN, YOLOX, and YOLOv7 networks. The two-stage transfer learning strategy employed significantly enhances the network's generalization ability and detection accuracy on the UAV inspection dataset. The proposed method provides a technical reference for fault detection of TLCs, which can potentially benefit the power transmission industry.