关键词:
水下图像增强
风格迁移
对抗学习
损失函数
摘要:
水下图像增强是完成水下任务的关键技术。针对水下图像存在的颜色失真、图像模糊的问题,本文设计一种基于领域自适应的水下图像增强算法(Underwater Domain Adaptation Network,UDA Net),实现无监督条件下水下图像的有效增强,显著改善原始水下图像的清晰度。该算法基于U-Net网络框架,利用卷积神经网络和多头注意力机制进行特征提取,引入对抗学习思想,在领域特征提取模块与输出模块中加入判别网络,同时优化源域特征增强损失、特征对齐损失以及输出对齐损失,保证源域到目标域的风格迁移与特征对齐,实现水下增强效果。此外,利用公开的水下数据集EUVP、UIEB以及UFO-120进行实验验证,并与前沿的增强算法进行对比实验,实验结果均表明了本文UDA Net算法的有效性,并在水下图像增强任务上具有良好的应用前景。