[发明专利]水下图像增强生成对抗网络的训练方法、介质及装置

摘要

本发明公开了一种水下图像增强生成对抗网络的训练方法、介质及装置，其中方法包括：获取原始图片，并将所述原始图片输入到水下图像失真模型，以生成所述原始图片对应的第一水下图片；对所述第一水下图片进行数据扩充，以生成多个第二水下图片，并根据所述原始图片和所述第二水下图片生成对应每个所述第二水下图片的训练数据对；根据所述训练数据对进行生成对抗网络的训练，以获取水下图像增强生成对抗网络。能够利用极小的数据集实现对水下图像进行增强，提高水下图像增强生成对抗网络的训练效率。