Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2410002(2022)

Infrared Image Recognition of Power Equipment Using Improved YOLOv4

Zhongxing Duan1、*, Yuming Zhang1, and Jiahao Ma2
Author Affiliations
  • 1College of Information and Control Engineering, Xi'an University of Architecture and Technology, Xi'an 710055, Shaanxi, China
  • 2State Grid Xi'an Power Supply Company, Xi'an 710032, Shaanxi, China
  • show less

    In this paper, we propose an improved YOLOv4 target detection model combined with an MSRCR image enhancement algorithm to mitigate the poor visual effect, large-scale difference, and unbalanced data category in infrared images of power equipment. First, we constructed a target detection dataset of infrared images of power equipment and applied the MSRCR algorithm to enhance the original infrared images and improve the low contrast and blur pixel of infrared images in rain and fog weather. This improves the detection ability of the model to power equipment in rain and fog weather. Second, a multi-scale convolution module was introduced in the YOLOv4 backbone network to obtain multi-scale features of input images using convolution kernels with different sizes to enhance the initial feature representation. To further improve the detection accuracy, the Focal loss function was used to solve the difficult problem of classification caused by unbalanced infrared image data. The test results show that the average recognition accuracy of the proposed method for eight power equipments is 96.31%, and the detection speed is 71 frame/s. The experimental results verify the effectiveness of the proposed method.

    Tools

    Get Citation

    Copy Citation Text

    Zhongxing Duan, Yuming Zhang, Jiahao Ma. Infrared Image Recognition of Power Equipment Using Improved YOLOv4[J]. Laser & Optoelectronics Progress, 2022, 59(24): 2410002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Sep. 6, 2021

    Accepted: Oct. 27, 2021

    Published Online: Oct. 31, 2022

    The Author Email: Duan Zhongxing (zhx_duan@163.com)

    DOI:10.3788/LOP202259.2410002

    Topics