Welcome to Journal of Graphics share: 

Journal of Graphics ›› 2024, Vol. 45 ›› Issue (1): 90-101.DOI: 10.11996/JG.j.2095-302X.2024010090

• Image Processing and Computer Vision • Previous Articles     Next Articles

IDD-YOLOv7: a lightweight method for multiple defect detection of insulators in transmission lines

ZHAI Yongjie(), ZHAO Xiaoyu, WANG Luyao, WANG Yaru(), SONG Xiaoke, ZHU Haoshuo   

  1. Department of Automation, North China Electric Power University, Baoding Hebei 071003, China
  • Received:2023-08-11 Accepted:2023-10-31 Online:2024-02-29 Published:2024-02-29
  • Contact: WANG Yaru (1990-), lecturer, Ph.D. Her main research interests cover pattern recognition, data mining, detection of transmission line components, etc. E-mail:wangyaru@ncepu.edu.cn
  • About author:

    ZHAI Yongjie (1972-), professor, Ph.D. His main research interest covers power vision. E-mail:zhaiyongjie@ncepu.edu.cn

  • Supported by:
    National Natural Science Foundation of China Joint Fund Project Key Support Project(U21A20486);Fundamental Research Funds for the Central Universities(2023JC006);National Natural Science Foundation Youth Fund project(62303184);Natural Science Foundation Youth Science Fund Project of Hebei Province(F2021502008)

Abstract:

The YOLO objective detection algorithm is currently the mainstream method for detecting insulator defects in image-based power transmission lines. However, due to the high complexity of existing models, a reasonable and effective parameter compression method is urgently needed as a prerequisite to establish the foundation for solving the dilemma of UAV edge device deployment. Additionally, the complex background of the insulator defect images captured by drones and small size of defects can lead to problems such as false detections and omissions. To address these issues, the Insulator Defect Detection-YOLOv7 (IDD-YOLOv7) model was proposed for multi-defect detection in power transmission line insulators, aiming to reduce model complexity and enhance robustness. Firstly, a coordinate attention mechanism was incorporated during the multi-scale feature fusion process to suppress interference from complex backgrounds and enhance the model’s global perception of small objects. Secondly, a C3GhostNetV2 module was designed to capture long-range dependencies between different spatial pixels, thus enhancing the model’s expressive power while reducing the parameter quantity and floating-point operation complexity. Lastly, the Focal-CIoU loss function was proposed to improve the contribution of high-quality anchors to the model and accelerate model convergence. Experimental results demonstrated that compared with the baseline model, the mAP50 of this method has increased by 3.8%, with precision and recall rates increasing by 1.7% and 7.6%, respectively, and the parameter quantity and floating-point operations have decreased by 18.3% and 14.0%, respectively. The AP50 of insulator self-explosion, damage, and flashover defects have increased by 0.8%, 4.5%, and 6.3%, respectively.

Key words: YOLOv7, insulator defect detection, attention mechanism, model complexity, lightweight, loss function

CLC Number: