Citation: | DENG Changzheng, LIU Mingze, FU Tian, GONG Mengqing, LUO Bingjie. Infrared Image Recognition of Substation Equipment Based on Improved YOLOv7-Tiny Algorithm[J]. Infrared Technology , 2025, 47(1): 44-51. |
To address the problem of low accuracy and slow recognition speed of infrared (IR) image target recognition of substation equipment in complex backgrounds, this study proposes an IR image recognition algorithm for substation equipment based on the improved YOLOv7-Tiny. First, the lightweight bottleneck structure GhostNetV2 bottleneck is introduced to replace a part of the CBS module and build a lightweight and efficient aggregation network known as a lightweight-efficient layer aggregation network. Simultaneously, a coordinate attention mechanism is embedded in the feature extraction stage to reduce the number of network parameters while strengthening the network's extraction of key features of the target and improving detection accuracy. The network coordinate loss function is replaced by SIoU_Loss to improve the anchor frame positioning accuracy and network convergence speed. The results show that the accuracy of the improved network is 96.28%, the detection rate is 26.42 frames/s, and the model size is reduced to 7.82 M. Compared with the original YOLOv7-Tiny algorithm, the detection rate is increased by 21.69%, the identification accuracy is improved, and the model size is reduced by 36.89%. These results meet the requirements of accurate real-time identification of substation equipment and lay a foundation for subsequent substation equipment fault diagnosis.
[1] |
王晶, 姚邹静, 赵春晖. 基于红外图像时空特征的电力设备故障诊断[J]. 控制工程, 2021, 28(8): 1683-1690.
WANG Jing, YAO Zoujing, ZHAO Chunhui. A fault diagnosis method for power equipment based on spatiotemporal features of infrared images[J]. Control Engineering of China, 2021, 28(8): 1683-1690.
|
[2] |
HAN S, YANG F, YANG G, et al. Electrical equipment identification in infrared images based on ROI-selected CNN method[J]. Electric Power Systems Research, 2020, 188: 106534. DOI: 10.1016/j.epsr.2020.106534
|
[3] |
吴添权, 郭竞, 苟先太, 等. 基于改进高斯卷积核的变电站设备红外图像检测方法[J]. 红外技术, 2021, 43(3): 230-236. http://hwjs.nvir.cn/cn/article/id/478e34f8-3f81-4419-8619-af60c5fb27f0
WU Tianquan, GUO Jing, GOU Xiantai, et al. Method of detecting substation equipment in infrared images based on improved gaussian convolution kernel[J]. Infrared Technology, 2021, 43(3): 230-236. http://hwjs.nvir.cn/cn/article/id/478e34f8-3f81-4419-8619-af60c5fb27f0
|
[4] |
DUAN K, BAI S, XIE L, et al. CenterNet: keypoint triplets for object detection[C]//IEEE International Conference on Computer Vision, 2019: 6569-6578.
|
[5] |
邝楚文, 何望. 基于红外与可见光图像的目标检测算法[J]. 红外技术, 2022, 44(9): 912-919. http://hwjs.nvir.cn/cn/article/id/60c5ef39-1d9c-4918-842f-3d86b939f3a6
KUANG Chuwen, HE Wang. Object detection algorithm based on infrared and visible light images[J]. Infrared Technology, 2022, 44(9): 912-919. http://hwjs.nvir.cn/cn/article/id/60c5ef39-1d9c-4918-842f-3d86b939f3a6
|
[6] |
王媛彬, 李媛媛, 段誉, 等. 基于轻量骨干网络和注意力结构的变电设备红外图像识别[J]. 电网技术, 2023, 47(10): 4358-4369.
WANG Yuanbin, LI Yuanyuan, DUAN Yu, et al. Infrared image recognition of power transformation equipment based on light backbone network and attention structure [J]. Power Grid Technology, 2023, 47(10): 4358-4369.
|
[7] |
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.
|
[8] |
HAN K, WANG Y, TIAN Q, et al. Ghostnet: More features from cheap operations[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020: 1580-1589.
|
[9] |
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.
|
[10] |
王有元, 李后英, 梁玄鸿, 等. 基于红外图像的变电设备热缺陷自调整残差网络诊断模型[J]. 高电压技术, 2020, 46(9): 3000-3007.
WANG Youyuan, LI Houying, LIANG Xuanhong, et al. Thermal defect self-tuning residual network diagnosis model of substation equipment based on infrared image[J]. High Voltage Technology, 2020, 46(9): 3000-3007.
|
[11] |
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.
|
[12] |
TANG Y, HAN K, GUO J, et al. GhostNetv2: enhance cheap operation with long-range attention[J]. Advances in Neural Information Processing Systems, 2022, 35: 9969-9982.
|
[13] |
HOU Q, ZHOU D, FENG J. Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2021: 13713-13722.
|
[14] |
LIN T Y, Goyal P, Girshick R, et al. Focal loss for dense object detection[C]//Proceedings of the IEEE International Conference on Computer Vision, 2017: 2980-2988.
|
[15] |
Gevorgyan Z. SIoU loss: More powerful learning for bounding box regression[J]. arXiv preprint arXiv: 2205.12740, 2022.
|
[16] |
Vaswani A, Shazeer N, Parmar N, et al. Attention is all you need[J/OL]. Advances in Neural Information Processing Systems, 2017, https://arxiv.org/abs/1706.03762.
|
[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 .
![]() |