[发明专利]基于条件生成对抗网络的水下图像增强方法及系统

权利要求书

1.一种基于条件生成对抗网络的水下图像增强方法，包括如下步骤：

步骤1：获取水下退化图像和对应的纯净图像的成对图像集，并分为训练集和测试集；

步骤2：将所有图像缩放为大小相同的尺寸；

步骤3：模型构建，包括：基于编解码器结构对图像进行全局和局部特征提取；将全局特征分别与各尺度的局部特征进行融合；将全局特征经过逐层上采样进行图像还原，每个上采样层与对应尺度的融合特征进行连接；将生成的图像送入判别器网络，判断其是否来自于真实数据，促使生成器网络进行调整；

步骤4：模型训练和测试，保存测试好的模型；

步骤5：利用测试好的模型对实际水下图像进行处理。

2.根据权利要求1所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，所述编解码器结构为改进的U-Net网络，其包括8层降采样，并通过逐层卷积的方式对输入图像进行全局和局部特征提取。

3.根据权利要求2所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，所述降采样层的每一个由LeakyRelu层、二维卷积层和批归一化层组成。

4.根据权利要求2所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，在降采样过程中基于SENet与NAM模块，将SENet中的全局平均池化模块更换为NAM模块中的批归一化比例因子，构建用于水下图像增强的注意力模块。

5.根据权利要求4所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，在所述注意力模块中，输入的特征图经过批归一化层和1×1卷积处理，并与权重系数相乘，再经过ReLU激活函数、1×1卷积层和sigmoid激活函数，最终与输入特征图进行跳层连接。

6.根据权利要求2所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，在逐层上采样的结果与具备相同分辨率的降采样结果进行跳层连接之前进行全局和局部特征融合，融合过程为：

步骤4-1：通过卷积核大小为1×1、步长为1的卷积层将全局特征f_g的通道数量c_g调整为与对应尺度i的局部特征图f_l相同的通道数c_i，该步骤表示为

f_g1＝F_conv(f_g，W)

其中，F_conv表示卷积操作，W为可学习权重；

步骤4-2：对f_g1进行复制，复制数量为h_i×w_i，其中，h_i与w_i为尺度i的局部特征图f_l的长度和宽度，该操作表示为

f_g2＝F_copy(f_g1，num＝h_i×w_i)

步骤4-3：将f_g2重塑为与f_l相同的维度h_i×w_i×c_i

f_g3＝F_re(f_g2，size＝h_i×w_i×c_i)

其中，F_re表示重塑操作。

步骤4-4：将f_g3与f_l进行连接操作

f_out＝F_concat(f_l，f_g3)。

7.根据权利要求2所述的一种基于条件生成对抗网络的水下图像增强方法，其特征在于，基于改进的U-Net网络进行图像还原，其包括8层上采样，并与降采样层对应。