基于VSA-UNet的电气设备紫外图像分割方法

本发明公开了一种基于VSA‑UNet的电气设备紫外图像分割方法，所述方法中，编码器使用VGG16Net的卷积层代替U‑Net网络的编码部分；将编码部分末端卷积层替换成空洞空间金字塔池化模块；在跳跃连接部分加入改进SENet模块，其包括压缩、激励和拼接三步操作。可以提高电气设备紫外图像分割精确度。从公开的电力设备紫外图像数据集选取紫外图像进行数据集制作，并将该数据集划分为训练集、验证集和测试集，然后对实验环境进行配置，并针对设备性能及紫外图像的特点设置训练参数，最后利用划分的数据集对模型进行训练、验证和测试。