一种基于生成对抗网络模型的水下图像修复方法和生成对抗网络模型训练方法

摘要

本发明提供了一种基于生成对抗网络模型的水下图像修复方法和生成对抗网络模型的训练方法，属于水下图像处理技术领域。生成对抗网络模型包括生成器网络和判别器网络，生成器网络包括依次设置的编码器网络块、特征转换网络块以及解码器网络块，生成器网络和判别器网络均包括多个BN层和PReLU层；在训练时，生成器网络中的编码器网络块和解码器网络块之间还设置有剩余路径网络块。本发明通过在编码器网络块和解码器网络块之间并添加了剩余路径网络块，增强生成器的细节特征提取能力，缓解了编码器低级特征与解码器高级特征之间的语义差异，避免了编码器下采样过程中的细节丢失，使得增强后的图像细节纹理更清晰。