• Infrared Technology
  • Vol. 46, Issue 6, 721 (2024)
Xiaodong WANG*, Tongfa LYU, Mingzheng BAO, Yongchun HE..., Peng XIN and Tao WU|Show fewer author(s)
Author Affiliations
  • [in Chinese]
  • show less
    DOI: Cite this Article
    WANG Xiaodong, LYU Tongfa, BAO Mingzheng, HE Yongchun, XIN Peng, WU Tao. Infrared Image Recognition Method for Power Equipment Based on Improved YOLO v5[J]. Infrared Technology, 2024, 46(6): 721 Copy Citation Text show less

    Abstract

    This study proposes an improved YOLO v5 method to solve the problems of inaccurate classification and insufficient feature extraction from power equipment infrared images. First, the visible light data and infrared images of the power equipment were fused using the transfer learning method. The triplet attention mechanism was then embedded into the feature extraction network for weighted intensification of key feature information. Finally, different targets were identified using multiscale fusion.The results show that compared with faster R-CNN and SSD, the proposed method has higher recognition accuracy and efficiency and is suitable for image recognition of multi-type power equipment in complex backgrounds. This method realizes a lightweight network model with a size of only 4.1 MB, which is a reduction of 80.8% compared to that of SSD, providing a novel and feasible scheme for intelligent infrared image detection of power equipment.
    WANG Xiaodong, LYU Tongfa, BAO Mingzheng, HE Yongchun, XIN Peng, WU Tao. Infrared Image Recognition Method for Power Equipment Based on Improved YOLO v5[J]. Infrared Technology, 2024, 46(6): 721
    Download Citation