基于多结构融合WGAN的模糊绝缘子图像复原方法研究  

A WGAN blur insulator image restoration method based on multi-structure fusion

作　　者：芦肇基 沈艳霞[1] 谭永强 LU Zhaoji;SHEN Yanxia;TAN Yongqiang(School of Internet of Things Engineering,Jiangnan University,Wuxi 214122,China;Nanjing Power Supply Company,State Grid Jiangsu Electric Power Co.,Ltd.,Nanjing 210000,China)

机构地区：[1]江南大学物联网工程学院,江苏无锡214122 [2]国网江苏省电力有限公司南京供电公司,江苏南京210000

出　　处：《电力系统保护与控制》2024年第22期166-175,共10页Power System Protection and Control

基　　金：国家自然科学基金项目资助(61573167,61572237)。

摘　　要：为解决因不可抗力因素导致无人机航拍绝缘子图像发生运动模糊的问题,提出一种基于多结构融合Wasserstein生成对抗网络(Wasserstein generate adversarial networks, WGAN)的模糊绝缘子图像复原方法。针对模糊图像复原问题,对基于Wasserstein距离的生成对抗网络加以改进,在损失函数中引入梯度惩罚项优化训练过程,保证模型训练的稳定性并提高图像复原质量。在生成网络中融入空洞卷积残差网络和卷积注意力机制,加强神经网络对图像有效特征的学习。实验结果表明,通过与其他算法比较,所提方法在峰值信噪比和结构相似度两种指标上均高于其他算法。对不同算法生成的图像进行比较,证明了该方法能有效提取图像细节特征,提高模糊绝缘子图像的复原质量。采用YOLOv5s目标检测算法进行实验,证明了所提方法对目标检测的准确率有所提升。There is a problem of motion blur caused by force majeure factors in the unmanned aerial vehicle aerial photography of insulator images.Thus a blur insulator image restoration method based on improved Wasserstein generative adversarial networks(WGAN)with multi structure fusion is proposed.An improved generative adversarial network based on Wasserstein distance is proposed to solve the problem of blur repair,and a gradient penalty is introduced into the loss function to optimize the training process.This ensures the stability of model training and improves the quality of image restoration.A dilated convolution residual network and convolutional attention mechanism are integrated into the generating network to strengthen the learning of effective features of images by the neural network.The results of experiment show that both the peak signal-to-noise ratio and the structural similarity index measure of the proposed method are higher than with other algorithms.The comparison of images generated by different algorithms proves that this method can effectively extract the detailed features and improve the quality of deblur image restoration.YOLOv5s is used for detection experiments.These demonstrate that the method presented enhances the accuracy of object detection.This work is supported by the National Natural Science Foundation of China(No.61573167 and No.61572237).

关 键 词：绝缘子图像复原 生成对抗网络 残差网络 卷积注意力机制 深度学习 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TM216[自动化与计算机技术—计算机科学与技术]