鲁棒的水下图像增强模型训练方法、图像增强方法及装置

本发明提供了一种鲁棒的水下图像增强模型训练方法、图像增强方法及装置，属于图像增强技术领域，该鲁棒的水下图像增强模型训练方法包括：采用对抗攻击策略根据原始水下图像生成对抗样本；其中，对抗攻击策略包括像素攻击和色彩偏移攻击中的至少一项；以原始水下图像和对抗样本为训练样本，以联合损失为损失函数对水下图像增强模型进行迭代训练，在满足最大迭代次数或者水下图像增强模型收敛的情况下，得到新的水下图像增强模型；其中，联合损失基于未参与对抗训练的水下图像增强模型的损失函数和对抗正则化项确定。本发明所述方法提高了水下图像增强模型对抗扰动的鲁棒性和图像增强性能，避免模型发生对抗扰动时生成恶化的图像。