一种基于生成对抗网络的变电站监控图像去雾方法技术

一种基于生成对抗网络的变电站监控图像去雾方法属于图像处理技术领域。本发明专利技术设计了多尺度特征提取模块、注意力模块和边缘增强模块；还设计了基于上述模块的生成网络，在对变电站监控图像去雾的同时，更好的保留图像边缘信息和颜色信息，使得去雾后的图像更清晰，更自然；设计了多尺度对抗网络，更好的提取特征信息，提高对抗网络的对图像是否是真实图像还是生成图像的判决能力，进而提高生成网络的图像去雾能力。采用本发明专利技术提出的方法对变电站监控图像进行去雾，可以有效的去处除图像中的雾霾，而且更够更好的保留图像的颜色信息和物体的边缘信息，去雾后的图像视觉效果更好。去雾后的图像视觉效果更好。去雾后的图像视觉效果更好。

【技术实现步骤摘要】
一种基于生成对抗网络的变电站监控图像去雾方法


[0001]本专利技术属于图像处理
，特别是涉及到一种基于生成对抗网络的变电站监控图像去雾方法。

技术介绍

[0002]变电站是电力系统的重要组成部分，它们通过将高压输电线路的电能转变为低压电能，并实现电能配送和调度，保障了电力系统的安全稳定运行。为此，需要定期对变电站进行巡检，确保设备安全可靠运行。通过巡检，可以发现潜在的故障和安全隐患，及时采取维护和修复措施，提高设备的运行效率和可用性，确保电力系统的稳定供应。传统变电站监控依赖于人工巡检，需要投入大量人力资源。然而，由于人力有限，难以实现对变电站的全面监控，导致部分区域存在盲区和漏检的风险。无人值守变电站视频监控系统通过安装摄像头和监控设备，实现对变电站各个角落的实时监控，弥补了人力不足的问题。然而雾霾天气由于大气粒子的散射作用，导致变电站室外监控摄像头拍摄到的图像模糊不清，难以为站内环境实时监测、人员安全智能评估、安全作业监控以及电子虚拟围栏等提供清晰图像，直接影响对监测的准确度，为此需要研究变电站图像去雾方法，提高雾霾天气下采集到的变电站图像质量。
[0003]变电站监控图像去雾一直是机器视觉领域的一个热门的技术问题。现有变电站监控图像去雾主要是分为基于图像增强的方法、基于物理模型的方法和基于深度学习的方法，基于图像增强的去雾方法是从图像的特征出发，通过利用或改进传统的增强算法来改善雾霾天气下图像的对比度和颜色信息，以此来达到去雾的目的，基于图像增强的方法主要关注去除雾霾并增强图像细节和对比度，但无法准确恢复真实场景的外观，去雾后的变电站图像失真较为严重。基于物理模型的图像去雾方法是通过建立光学和大气散射模型来对雾霾进行建模和估计，并利用这些模型进行图像去雾，当模型参数比较准确的时候，该方法可以达到很好的去雾效果，但是在实际应用过程中很难估计出适合所有场景的准确的模型参数，因此，此类方法无法适用于不同场景下的变电站图像去雾。基于深度学习的图像去雾方法能够学习复杂的图像特征和雾霾去除的映射关系，从而提供更准确、自然的去雾结果。虽然基于深度学习的去雾方法相对其他两类方法具有较好的去雾效果，但是去雾后的图像仍然存在一定颜色失真，过饱和边缘信息丢失严重的问题。
[0004]因此现有技术当中亟需要一种新型的技术方案来解决这一问题。

技术实现思路

[0005]本专利技术所要解决的技术问题是：提供一种基于生成对抗网络的变电站监控图像去雾方法用于解决变电站监控图像去雾后，图像颜色失真，过饱和以及边缘信息丢失严重的技术问题。
[0006]一种基于生成对抗网络的变电站监控图像去雾方法，包括以下步骤，并且以下步骤顺次进行:
[0007]S1：构建变电站监控图像去雾方法训练和测试所需的数据集；
[0008]S2：构建生成对抗网络中的生成网络，用于变电站监控图像去雾；
[0009]S3：构建生成对抗网络中的对抗网络，用于判断输入的图像是真实的无雾图像还是生成的无雾图像；
[0010]S4：设计损失函数，用来衡量生成对抗网络的性能；
[0011]S5：模型训练：对由步骤S2和S3组成的生成对抗网络进行训练和测试，通过不断的循环迭代，优化步骤S4所设计的损失函数，得到最优的生成网络和对抗网络模型参数,生成对抗网络模型训练完成；
[0012]S6：模型测试：对步骤S5训练得到的生成对抗网络模型进行测试，测试指标满足设定的阈值，则将步骤S5已训练的模型作为最终去雾模型；否则返回步骤S5，调整参数继续进行模型训练；
[0013]S7：模型应用：将雾霾天气下采集到的变电站监控图像输入到最终去雾模型中，得到去雾后的变电站监控图像。
[0014]所述步骤S1具体包括：从变电站监控系统中的选择不同背景和光照度的无雾图像和含雾图像各10000张，并将其进行裁剪成像素大小一致的图像，根据大气散射模型对裁剪后的无雾图像进行加雾处理，雾的浓度随机产生，得到10000张合成的含雾图像；裁剪后的无雾图像和对应的合成后的含雾图像组成图像对，随机选择8000对图像作为训练集，剩下的2000对图像作为测试集1，裁剪后的10000张含雾图像作为测试集2。
[0015]所述步骤S2具体包括：
[0016]用于变电站监控图像去雾的生成网络由：编码器，解码器，4个级联的空洞卷积，边缘检测以及边缘增强模块组成；训练集中的各含雾图像经过边缘检测得到的图像F0，将图像F0与该含雾图像进行拼接操作得到图像F1，图像F1作为生成网络的输入图像；所述F1图像输入到编码器中，在编码器中依次经过带有LeakyReLU激活函数的3
×
3卷积、多尺度特征提取模块、注意力模块和下采样操作的处理，得到特征图F2，再对所述特征图F2依次经过多尺度特征提取模块，注意力模块和下采样操作的处理得到特征图F3，再对所述特征图F3经过多尺度特征图提取模块，得到编码器的输出特征图F4；对所述特征图F4输入至4个级联的空洞卷积中进行连续4次空洞卷积操作，得到特征图F5；特征图F5作为输入特征图输入到解码器中，对所述特征图F5进行上采样操作,得到特征图F6，再对特征图F6与特征图F3进行拼接得到特征图F7，对所述特征图F7依次经过1
×
1卷积、注意力模块、多尺度特征提取模块和上采样操作的处理，得到特征图F8，再对特征图F8与特征图F2进行拼接得到特征图F9，再对所述特征图F9进行依次经过1
×
1卷积、注意力模块、多尺度特征提取模块和Tanh激活函数的3
×
3卷积的处理，得到图F10，特征图F10也是解码器的输出图；将所述特征图F0输入至边缘增强模块依次进行下采样操作、3个连续的含有LeakyReLU激活函数的3
×
3卷积操作和1个下采样操作，得到图F11，图F11就是边缘增强模块的输出图；图F11与图F10进行元素相加得到生成网络的输出图像，也就是变电站监控图像去雾后的图像。
[0017]所述多尺度特征提取模块由并行四个分支组成，四个分支的输入特征图相同；第一个分支由1个1
×
1卷积组成；第二个分支由三个并列的空洞率分别为1、2和4的3
×
3空洞卷积组成，三个空洞卷积的输出特征图通过拼接操作得到新的输出特征图，新的输出特征图依次经过1个1
×
1卷积，1个3
×
3卷积，2个连续的带有LeakyReLU激活函数的3
×
3卷积，得
到第二个分支的输出特征图；第三个分支依次由平均池化，3个连续的带有LeakyReLU激活函数的3
×
3卷积和一个上采样操作组成；第四个分支依次由最大池化，3个连续的带有LeakyReLU激活函数的3
×
3卷积和一个上采样操作组成；最后，四个分支的输出特征图经过元素相加操作得到该多尺度特征提取模块的输出特征图。
[0018]所述注意力模块包含2个分支，两个分支的输入特征图均是注意力模本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于生成对抗网络的变电站监控图像去雾方法，其特征是：包括以下步骤，并且以下步骤顺次进行:S1：构建变电站监控图像去雾方法训练和测试所需的数据集；S2：构建生成对抗网络中的生成网络，用于变电站监控图像去雾；S3：构建生成对抗网络中的对抗网络，用于判断输入的图像是真实的无雾图像还是生成的无雾图像；S4：设计损失函数，用来衡量生成对抗网络的性能；S5：模型训练：对由步骤S2和S3组成的生成对抗网络进行训练和测试，通过不断的循环迭代，优化步骤S4所设计的损失函数，得到最优的生成网络和对抗网络模型参数,生成对抗网络模型训练完成；S6：模型测试：对步骤S5训练得到的生成对抗网络模型进行测试，测试指标满足设定的阈值，则将步骤S5已训练的模型作为最终去雾模型；否则返回步骤S5，调整参数继续进行模型训练；S7：模型应用：将雾霾天气下采集到的变电站监控图像输入到最终去雾模型中，得到去雾后的变电站监控图像。2.根据权利要求1所述的一种基于生成对抗网络的变电站监控图像去雾方法，其特征是：所述步骤S1具体包括：从变电站监控系统中的选择不同背景和光照度的无雾图像和含雾图像各10000张，并将其进行裁剪成像素大小一致的图像，根据大气散射模型对裁剪后的无雾图像进行加雾处理，雾的浓度随机产生，得到10000张合成的含雾图像；裁剪后的无雾图像和对应的合成后的含雾图像组成图像对，随机选择8000对图像作为训练集，剩下的2000对图像作为测试集1，裁剪后的10000张含雾图像作为测试集2。3.根据权利要求2所述的一种基于生成对抗网络的变电站监控图像去雾方法，其特征是：所述步骤S2具体包括：用于变电站监控图像去雾的生成网络由：编码器，解码器，4个级联的空洞卷积，边缘检测以及边缘增强模块组成；训练集中的各含雾图像经过边缘检测得到的图像F0，将图像F0与该含雾图像进行拼接操作得到图像F1，图像F1作为生成网络的输入图像；所述F1图像输入到编码器中，在编码器中依次经过带有LeakyReLU激活函数的3
×
3卷积、多尺度特征提取模块、注意力模块和下采样操作的处理，得到特征图F2，再对所述特征图F2依次经过多尺度特征提取模块，注意力模块和下采样操作的处理得到特征图F3，再对所述特征图F3经过多尺度特征图提取模块，得到编码器的输出特征图F4；对所述特征图F4输入至4个级联的空洞卷积中进行连续4次空洞卷积操作，得到特征图F5；特征图F5作为输入特征图输入到解码器中，对所述特征图F5进行上采样操作,得到特征图F6，再对特征图F6与特征图F3进行拼接得到特征图F7，对所述特征图F7依次经过1
×
1卷积、注意力模块、多尺度特征提取模块和上采样操作的处理，得到特征图F8，再对特征图F8与特征图F2进行拼接得到特征图F9，再对所述特征图F9进行依次经过1
×
1卷积、注意力模块、多尺度特征提取模块和Tanh激活函数的3
×
3卷积的处理，得到图F10，特征图F10也是解码器的输出图；将所述特征图F0输入至边缘增强模块依次进行下采样操作、3个连续的含有LeakyReLU激活函数的3
×
3卷积操作和1个下采样操作，得到图F11，图F11就是边缘增强模块的输出图；图F11与图F10进行元素相加得到生成网络的输出图像，也就是变电站监控图像去雾后的图像。
4.根据权利要求3所述的一种基于生成对抗网络的变电站监控图像去雾方法，其特征是：所述多尺度特征提取模块由并行四个分支组成，四个分支的输入特征图相同；第一个分支由1个1
×
1卷积组成；第二个分支由三个并列的空洞率分别为1、2和4的3
×
3空洞卷积组成，三个空洞卷积的输出特征图通过拼接操作得到新的输出特征图，新的输出特征图依次经过1个1
×
1卷积，1个3
×
3卷积，2个连续的带有LeakyReLU激活函数的3
×
3卷积，得到第二个分支的输出特征图；第三个分支依次由平均池化，3个连续的带有LeakyReLU激活函数的3
×
3卷积和一个上采样操作组成；第四个分支依次由最大池化，3个连续的带有LeakyReLU激活函数的3
×
3卷积和一个上采样操作组成；最后，四个分支的输出特征图经过元素相加操作得到该多尺度特征提取模块的输出特征图。5.根据权利要求3所述的一种基于生成对抗网络的变电站监...

【专利技术属性】
技术研发人员：贾雁飞，郭铁滨，刘赟静，于文硕，关潇卓，王光璞，刘旭，付饶，杨烁，李达，石根华，姜科宇，
申请(专利权)人：国网吉林省电力有限公司吉林供电公司，
类型：发明
国别省市：