基于生成对抗网络的水下图像增强方法、系统及电子设备

本申请公开了一种基于生成对抗网络的水下图像增强方法、系统及电子设备，本方法通过获取水下图像数据集和陆地图像数据集；根据水下图像数据集和陆地图像数据集，确定训练集；构建包含多个分支的多尺度并行编码器，以及，构建包含多尺度特征密集融合模块的多尺度融合解码器；构建包含两个生成器和两个判别器的基于生成对抗网络的水下图像增强模型，其中，每个生成器基于多尺度并行编码器和多尺度融合解码器构建；采用训练集对基于生成对抗网络的水下图像增强模型进行训练，得到训练好的水下图像增强模型，以便采用训练好的水下图像增强模型对待增强的水下图像进行图像增强。本申请能够提高图像增强效果，并提高图像增强的准确度。