Citation: | 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-80. |
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.
[1] |
吴慧海, 沈文忠. 基于TA-YOLO的电力设备红外图像检测方法[J]. 信息技术与信息化, 2022(3): 17-20.
WU Huihai, SHEN Wenzhong. TA-YOLO based infrared image detection method for power equipment [J]. Information Technology and Informatization, 2022(3): 17-20.
|
[2] |
赵庆生, 王雨滢, 梁定康, 等. 基于BOF图像检索算法的变电站设备图像分类[J]. 激光与光电子学进展, 2020, 57(18): 138-146.
ZHAO Qingsheng, WANG Yuying, LIANG Dingkang, et al. Image classification of substation equipment based on BOF image retrieval algorithm[J]. Advances in Laser and Optoelectronics, 2020, 57(18): 138-146.
|
[3] |
JIANG A, YAN N, SHEN B, et al. Research on infrared image recognition method of power equipment based on deep learning[C]//2020 IEEE International Conference on High Voltage Engineering and Application (ICHVE), 2020: 1-4.
|
[4] |
裴泽林, 赵曙光, 王建强. 基于Meta-Network的变压器红外图像分类方法研究[J]. 科技风, 2021(5): 87-88.
PEI Zelin, ZHAO Shuguang, WANG Jianqiang. Research on meta-network based transformer infrared image classification method[J]. Science and Technology Wind, 2021(5): 87-88.
|
[5] |
周可慧, 廖志伟, 肖异瑶, 等. 基于改进CNN的电力设备红外图像分类模型构建研究[J]. 红外技术, 2019, 41(11): 1033-1038. https://www.opticsjournal.net/Articles/OJ407b66a963da9993/Abstract
ZHOU Kehui, LIAO Zhiwei, XIAO Yiyao, et al. Research on constructing infrared image classification model for power equipment based on improved CNN[J]. Infrared Technology, 2019, 41(11): 1033-1038. https://www.opticsjournal.net/Articles/OJ407b66a963da9993/Abstract
|
[6] |
段中兴, 张雨明, 马佳豪. 基于改进YOLOv4的电力设备红外图像识别[J]. 激光与光电子学进展, 2022, 59(24): 2410002-2410002-8.
DUAN Z X, ZHANG Y M, MA J H. Improved YOLOv4-based infrared image recognition for power equipment[J]. Laser & Optoelectronics Progress, 2022, 59(24): 2410002-2410002-8.
|
[7] |
CONG S, PU H, WANG X, et al. Application of improved YOLOv5 in infrared image recognition of electrical equipment[C]//2023 8th Asia Conference on Power and Electrical Engineering (ACPEE), 2023: 1836-1843.
|
[8] |
LI J, XU Y, NIE K, et al. PEDNet: a lightweight detection network of power equipment in infrared image based on YOLOv4-Tiny[J]. IEEE Transactions on Instrumentation and Measurement, 2023, 72: 1-12.
|
[9] |
余一聪, 何领朝, 蔡荣贵, 等. 基于改进YOLOv7的绝缘子缺失检测方法[J]. 黑龙江工业学院学报: 综合版, 2023, 23(7): 83-90.
YU Yicong, HE Lingzhao, CAI Ronggui, et al. Insulator missing detection method based on improved YOLOv7[J]. Journal of Heilongjiang Institute of Technology: General Edition, 2023, 23(7): 83-90.
|
[10] |
WANG C Y, Bochkovskiy A, LIAO H Y M. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023: 7464-7475.
|
[11] |
JIANG P, Ergu D, LIU F, et al. A review of Yolo algorithm developments[J]. Procedia Computer Science, 2022, 199: 1066-1073.
|
[12] |
Roy A M, Bhaduri J, Kumar T, et al. WilDect-YOLO: An efficient and robust computer vision-based accurate object localization model for automated endangered wildlife detection[J]. Ecological Informatics, 2023, 75: 101919.
|
[13] |
Redmon J, Divvala S, Girshick R, et al. You only look once: Unified, real-time object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2016: 779-788.
|
[14] |
Redmon J, Farhadi A. YOLO9000: better, faster, stronger[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2017: 7263-7271.
|
[15] |
Redmon J, Farhadi A. YOLOv3: An incremental improvement[J]. arXiv preprint arXiv: 1804.02767, 2018.
|
[16] |
Bochkovskiy A, WANG C Y, LIAO H Y M. YOLOv4: Optimal speed and accuracy of object detection[J]. arXiv preprint arXiv: 2004.10934, 2020.
|
[17] |
Olorunshola O E, Irhebhude M E, Evwiekpaefe A E. A comparative study of YOLOv5 and YOLOv7 object detection algorithms[J]. Journal of Computing and Social Informatics, 2023, 2(1): 1-12.
|
[18] |
Woo S, Park J, Lee J Y, et al. Cbam: Convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision (ECCV), 2018: 3-19.
|
[19] |
WANG Xiaorong, XU Yan, ZHOU Jianping, et al. Safflower picking recognition in complex environments based on an improved YOLOv7[J]. Transactions of the Chinese Society of Agricultural Engineering, 2023, 39(6): 169-176.
|
[20] |
HU J, SHEN L, SUN G. Squeeze-and-excitation networks[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2018: 7132-7141.
|
[21] |
WANG Q, WU B, ZHU P, et al. ECA-Net: Efficient channel attention for deep convolutional neural networks[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020: 11534-11542.
|
[1] | ZHANG Fei, WANG Jian, ZHANG Yuesong. Lightweight Infrared Small Target Detection Algorithm under Oblique View Based on YOLOv5[J]. Infrared Technology , 2025, 47(2): 217-225. |
[2] | ZHANG Xingping, SHAO Yanhua, MEI Yanying, ZHANG Xiaoqiang, CHU Hongyu. Aerial Infrared Pedestrian Detection with Saliency Map Fusion[J]. Infrared Technology , 2024, 46(9): 1043-1050. |
[3] | CHEN Sijing, FU Zhitao, LI Ziqian, NIE Han, SONG Jiawen. A Visible and Infrared Image Fusion Algorithm Based on Adaptive Enhancement and Saliency Detection[J]. Infrared Technology , 2023, 45(9): 907-914. |
[4] | LONG Zhiliang, DENG Yueming, WANG Runmin, DONG Jun. Infrared and Visible Image Fusion Based on Saliency Detection and Latent Low-Rank Representation[J]. Infrared Technology , 2023, 45(7): 705-713. |
[5] | LI Shiji, LI Zhongmin, LI Wei. Pedestrian Detection Method Based on Improved ViBe and YOLO v3 Algorithms[J]. Infrared Technology , 2023, 45(2): 137-142. |
[6] | LAI Benlin, ZHANG Yongan, ZHANG Yaping, TIAN Li, ZHAO Danlu, HUANG Junhao, WANG Gang, ZHAO Hang. Research on Stress Field Detection of Building Blasting Environment Based on Infrared Digital Holography[J]. Infrared Technology , 2023, 45(1): 102-109. |
[7] | YE Kuntao, LI Wen, SHU Leilei, LI Sheng. Infrared and Visible Image Fusion Method Based on Improved Saliency Detection and Non-subsampled Shearlet Transform[J]. Infrared Technology , 2021, 43(12): 1212-1221. |
[8] | ZHAO Shuang, CHEN Shuyue, WANG Qiaoyue. Infrared Pedestrian Detection in Complex Night Scenes[J]. Infrared Technology , 2021, 43(6): 575-582. |
[9] | CHE Kai, XIANG Zhengtao, CHEN Yufeng, LYU Jian, ZHOU Yun. Research on Infrared Image Pedestrian Detection Based on Improved Fast R-CNN[J]. Infrared Technology , 2018, 40(6): 578-584. |
[10] | WEI Li, DING Meng, ZENG Lijun. Pedestrian Detection Based on Objectness and Sparse Coding in a Single Infrared Image[J]. Infrared Technology , 2016, 38(9): 752-757. |
1. |
相敏月,涂振宇,孙逸飞,方强,马飞. 基于ECA和BIFPN的低照度环境下的行人目标检测算法. 智能计算机与应用. 2023(09): 189-193 .
![]() |