首页> 中文期刊> 《水下无人系统学报》 >基于UNDERWATER-CUT模型的水下图像增强算法

基于UNDERWATER-CUT模型的水下图像增强算法

         

摘要

针对水下图像的颜色失真和对比度失衡问题,提出了一种基于UNDERWATER-CUT模型的弱监督水下图像增强算法。该算法网络训练时无需成对训练集,通过图像分块的方式构建对比学习正负样本,约束了图像生成的内容。使用了结构相似性损失函数对水下图像增强进行约束,确保CUT模型在对水下图像域到脱水图像域进行转换过程中的物体结构不变。同时还使用alignment和uniformity 2个简单函数的线性组合来近似逼近CUT模型的InfoNCE损失函数,使得改进后的模型训练更容易收敛到最优值。实验结果证明经过文中算法增强后的图像,颜色失真得到极大的修正,图像中的物体结构和修复前的图像基本一致。

著录项

相似文献

  • 中文文献
  • 外文文献
  • 专利
获取原文

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号