Laser & Infrared, Volume. 55, Issue 3, 399(2025)
An accurate segmentation method for infrared image of electrical equipment under complex environment
The precise segmentation of electrical equipment is a key step in infrared image fault diagnosis. An accurate segmentation method is proposed for infrared images of electrical equipment in complex backgrounds to address the issue of detail loss with mainstream semantic segmentation methods. Firstly, the PSPNet is improved by incorporating UNet network as the main structure to decode the multi-scale pyramid pooling of features extracted by UNet's top layer. Secondly, Convolutional Block Attention Mechanism (CBAM) is integrated into the feature extraction backbone network to incorporate channel and spatial attention mechanisms for gathering image context information from both dimensions, enhancing the network's focus on electrical equipment to improve its anti-interference capability. Finally, the PSPnet-CBAM-Unet network is constructed, and the features output by the CBAM are used as inputs for lower-level feature extraction and skip-connection features in the decoding layer. The effectiveness of this paper's method is tested with the segmentation of three types of devices in infrared images under complex backgrounds including voltage transformers, current transformers, and circuit breakers. Experimental results demonstrate that the proposed method achieves intersection over union and accuracy greater than 92% and 94% respectively, and the accuracy of segmentation is better than that of UNet, PSPNet, and Deeplabv3+ networks, and it is more accurate for the detail segmentation of infrared images of electrical equipment in a complex background.
Get Citation
Copy Citation Text
WANG Qi, ZHANG Xin-wei, TONG Yue, WANG Yu-qing, ZHANG Jin, WANG Yong-tao, YUAN Xiao-cui. An accurate segmentation method for infrared image of electrical equipment under complex environment[J]. Laser & Infrared, 2025, 55(3): 399
Category:
Received: Apr. 29, 2024
Accepted: Apr. 23, 2025
Published Online: Apr. 23, 2025
The Author Email: