Infrared Technology, Volume. 47, Issue 1, 72(2025)

IR Image Classification and Detection of Power Equipment Based on CBAM Improvement

Jia CHEN1,2, Chengbo YU1,2、*, Shibing WANG2,3, Qichao JIANG1,2, Xin HE1,2, and Wei ZHANG1,2
Author Affiliations
  • 1School of Electrical and Electronic Engineering, Chongqing University of Technology, Chongqing 400054, China
  • 2Chongqing Energy Internet Engineering Technology Research Center, Chongqing 400054, China
  • 3State Grid Chongqing Electric Power Company Shinan Power Supply Branch, Chongqing 401336, China
  • show less

    To address the problems of complicated data and low detection accuracy for deep-learning target detection of IR images of power equipment in complex environments, this study proposes a convolutional block attention module (CBAM) based on YOLOv7 to improve the classification algorithm for IR images of power equipment. First, the existing dataset is labeled and divided into training, validation, and test sets in a certain proportion and then introduced into the backbone network of YOLOv7 to enable the model to emphasize the region of interest and suppress useless information. Second, the divided dataset is put into the improved YOLOv7 for model training, and six improved YOLOv5s models are compared. The experimental results show that the improved YOLOv7 model outperforms YOLOv7, YOLOv5s, and six attention models based on YOLOv5s under the same experimental conditions. The improved YOLOv7 exhibits significantly improved performance and achieves fast and accurate IR image classification.

    Tools

    Get Citation

    Copy Citation Text

    CHEN Jia, YU Chengbo, WANG Shibing, JIANG Qichao, HE Xin, ZHANG Wei. IR Image Classification and Detection of Power Equipment Based on CBAM Improvement[J]. Infrared Technology, 2025, 47(1): 72

    Download Citation

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

    Category:

    Received: Sep. 6, 2023

    Accepted: Feb. 18, 2025

    Published Online: Feb. 18, 2025

    The Author Email: Chengbo YU (yuchengbo@cqut.edu.cn)

    DOI:

    CSTR:32186.14.

    Topics