Hunan Electric Power ›› 2025, Vol. 45 ›› Issue (1): 136-143.doi: 10.3969/j.issn.1008-0198.2025.01.020

• Artifical Intelligence and Digitizatrion in Electrical Power • Previous Articles     Next Articles

Light Weight Visual Recognition Method of Small Target Hardware for Transmission Lines Based on Improved YOLO Algorithm

ZOU Dehua1,2, ZHANG Hongwei3, JIANG Wei3, GONG Chuang3   

  1. 1. Hunan Province Key Laboratory of Intelligent Live Working Technology and Equipment(ROBOT) (State Grid Hunan Extra-High Voltage Transmission Company), Changsha 420100, China;
    2. Live Inspection and Intelligent Operation Technology State Grid Corporation Laboratory (State Grid Hunan Extra-High Voltage Transmission Company), Changsha 420100, China;
    3. School of Mechanical Engineering and Automation, Wuhan Textile University, Wuhan 430073, China
  • Received:2024-09-26 Revised:2024-11-19 Online:2025-02-25 Published:2025-03-05

Abstract: Aiming at the problem that deep learning-based transmission line hardware detection methods can not ensure the lightweight model while improving the detection accuracy of small targets, a lightweight detection method for small target hardware on transmission lines is proposed. Based on YOLOv4, the method firstly combines density-based spatial clustering of applications with noise and the anchor frame optimization strategy with multiple K-means to optimize the preset frame selection according to the characteristics of the small target data, speed up the network convergence and improve the detection accuracy. Secondly, GhostnetV2 is used to achieve model light weighting. Then the network of small object enhanced multi-scale detection network, is designed to enhance the feature extraction ability, strengthen the fusion of shallow and deep features, and solve the problem of feature loss caused by down sampling cross-step convolution. Finally, focal loss function is used to optimize sample allocation and depth-wise separable convolution is used to reduce model complexity. The mean average precision(mAP) of this algorithm on the self-constructed small hardwaredataset reaches 64.73%, and the model computation and parameter counts are reduced by 78% and 71% compared to the pre-improvement period, while there is almost no loss of detection accuracy. The mAP reaches 38.43% on the public dataset VisDrone2019, which gives the algorithm a superiorsmall target detection performance compared to other algorithms.

Key words: transmission line hardware, density clustering, small target detection, light-weighting, feature fusion, focal loss function

CLC Number: