Infrared Technology, Volume. 47, Issue 6, 770(2025)

UV Image Discharge Spot Segmentation for Electrical Equipment Based on Improved U-net

Wanke SHEN1, Luojingyi LI1, Chunhua FANG1、*, Quancai JIANG1, Jiewei LU2, Xingyu XIA1, and Wanzhao PENG1
Author Affiliations
  • 1College of Electrical Engineering & Renewable Energy, Chain Three Gorges University, Yichang 443002, China
  • 2Kaihua Power Supply Company of State Grid Zhejiang Electric Power Company, Quzhou 324000, China
  • show less

    This paper proposes a semantic segmentation model called VA-Unet, designed to address the challenges of complex backgrounds, slight spot separation, complex feature selection, and low segmentation accuracy encountered in ultraviolet (UV) detection tasks of electrical equipment. VA-Unet incorporates the VGG16 feature extraction module and transfer learning to accelerate training and enhance the model's generalization capability. Additionally, an Attention Gate is integrated to improve segmentation precision by focusing on relevant features, enabling accurate detection of UV discharge spots in images. To address the issue of sample imbalance in the UV discharge spot dataset, VA-Unet employs a hybrid loss function in place of a conventional single loss function. Experimental results demonstrate that VA-Unet achieves superior performance in the precise localization and accurate segmentation of UV discharge spots. The model attains an IoU of 84.09%, PA of 88.20%, and F1-score of 91.35%, representing improvements of 14.41%, 3.24%, and 9.22%, respectively, compared to the baseline U-Net model.

    Tools

    Get Citation

    Copy Citation Text

    SHEN Wanke, LI Luojingyi, FANG Chunhua, JIANG Quancai, LU Jiewei, XIA Xingyu, PENG Wanzhao. UV Image Discharge Spot Segmentation for Electrical Equipment Based on Improved U-net[J]. Infrared Technology, 2025, 47(6): 770

    Download Citation

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

    Category:

    Received: Aug. 28, 2023

    Accepted: Jul. 3, 2025

    Published Online: Jul. 3, 2025

    The Author Email: FANG Chunhua (1801238638@qq.com)

    DOI:

    Topics