[发明专利]一种架空输电线路视频监控图像去雾方法及装置

权利要求书

1.一种架空输电线路视频监控图像去雾方法，其特征在于，包括：

获取若干数量的架空输电线路的含雾图像和不含雾图像；

构建结构相同的第一生成网络和第二生成网络；

构建结构相同的第一对抗网络和第二对抗网络；

采用所述第一生成网络、所述第一对抗网络和所述第二生成网络生成第一生成对抗网络；

采用所述第二生成网络、所述第二对抗网络和所述第一生成网络生成第二生成对抗网络；

采用所述含雾图像训练所述第一生成对抗网络，采用所述不含雾图像训练所述第二生成对抗网络，得到目标生成网络；

将待去雾架空输电线路视频监控图像输入所述目标生成网络，得到去雾图像。

2.根据权利要求1所述的方法，其特征在于，所述第一生成网络包括编码器和解码器；所述将待去雾架空线路输电线路视频监控图像输入所述目标生成网络，得到去雾图像的步骤，包括：

将所述待去雾架空输电线路视频监控图像输入所述编码器，生成输出特征图；

将所述输出特征图输入所述解码器中，生成去雾图像。

3.根据权利要求2所述的方法，其特征在于，所述编码器包括第一卷积层、第一批归一化层、ReLU激活函数层、第一模块组和多尺度模块；所述将所述待去雾架空输电线路视频监控图像输入所述编码器，生成输出特征图的步骤，包括：

通过所述第一卷积层、所述第一批归一化层和所述ReLU激活函数层对所述待去雾架空输电线路视频监控图像进行特征提取，得到第一特征图；

将所述第一特征图输入级联的所述第一模块组中，得到第二特征图；

将所述第二特征图输入到所述多尺度模块进行深度提取，得到输出特征图。

4.根据权利要求3所述的方法，其特征在于，所述第一模块组包括多尺度特征提取模块、下采样操作层和第二卷积层；所述将所述第一特征图输入级联的所述第一模块组中，得到第二特征图的步骤，包括：

通过所述多尺度特征提取模块从所述第一特征图中提取低频特征图；

将所述低频特征图输入所述下采样操作层生成非线性特征图；

将所述非线性特征图输入所述第二卷积层，生成第二特征图。

5.根据权利要求2所述的方法，其特征在于，所述解码器包括第二模块组、第三卷积层、第二批归一化层和Tanh激活函数层；所述将所述输出特征图输入所述解码器中，生成去雾图像的步骤，包括：

将所述输出特征图输入级联的第二模块组中，得到第三特征图；

通过第三卷积层、第二批量归一化层和Tanh激活函数层将所述第三特征图转化为去雾图像。

6.根据权利要求1所述的方法，其特征在于，所述采用所述含雾图像训练所述第一生成对抗网络，采用所述不含雾图像训练所述第二生成对抗网络，得到目标生成网络的步骤，包括：

将所述含雾图像输入所述第一生成对抗网络，得到第一输出结果；

将所述不含雾图像输入所述第二生成对抗网络，得到第二输出结果；

根据所述第一输出结果、所述第二输出结果和预设损失函数计算损失；

根据所述损失调整所述第一生成对抗网络和所述第二生成对抗网络的参数，直至所述损失达到最小时，将对应的完成训练的第一生成网络确定为目标生成网络。