改进U-net的电气设备紫外图像放电光斑分割

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

  • 摘要: 提出了一种名为VA-Unet的语义分割模型,旨在解决传统分割方法在电气设备紫外检测任务中面临的复杂背景及小光斑分离困难、特征选取复杂、分割精准度低等问题。VA-Unet引入了VGG16特征提取模块和迁移学习,提高训练速度并增强模型泛化能力;同时,增加了注意力门(Attention Gate)以提高模型精度,从而实现对紫外图像放电光斑的精准分割。此外,VA-Unet采用混合损失函数代替单一损失函数,解决了紫外放电光斑数据集样本不平衡的问题。实验表明,VA-Unet模型在紫外图像放电光斑的精准定位和准确分割方面表现突出,其IoU,PA,F1-score评价指标分别达到84.09%,88.20%,91.35%,相较于初始U-net网络,分别提升了14.41%,3.24%,9.22%。

     

    Abstract: 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.

     

/

返回文章
返回