电子测量技术2023，Vol.46Issue(1) ：181-187.DOI:10.19651/j.cnki.emt.2210291

双注意力机制与改进U-Net的水下图像增强

Underwater image enhancement based on dual attention mechanism and improved U-Net

王海涛 林森 陶志勇
电子测量技术2023，Vol.46Issue(1) ：181-187.DOI:10.19651/j.cnki.emt.2210291
✕
  • NETL
  • NSTL
  • 维普
  • 万方数据

双注意力机制与改进U-Net的水下图像增强

Underwater image enhancement based on dual attention mechanism and improved U-Net

王海涛 1林森 1陶志勇2
扫码查看

作者信息

  • 1. 沈阳理工大学信息科学与工程学院 沈阳110159
  • 2. 辽宁工程技术大学电子与信息工程学院 葫芦岛125105
  • 折叠

摘要

针对现有的水下增强算法存在色彩失真和去雾效果不好等问题,本文提出基于双注意力机制与改进U-Net的水下图像增强算法.首先采用颜色校正模块对红、绿、蓝三通道进行处理,减少色偏的影响;然后将通道注意力、空间注意力与U-Net网络相融合,对颜色校正后的图像进行去雾、去噪等处理,保留图像纹理细节的同时,实现对比度的增强;最后采用金字塔融合模块将不同分辨率的图像特征进行融合,获得视觉上清晰的图像.实验结果表明,基于UIEBD和UFO-120测试集,UCIQE、NIQE、SURF以及信息熵的平均值分别为0.608 1、4.440 3、31.5和7.649 5,所提算法在主观视觉质量和客观评价指标上都优于其他经典及新颖算法,增强后水下图像去雾效果良好且在颜色校正方面也具有明显优势,显著提高了水下图像的视觉质量.

关键词

图像处理/注意力机制/U-Net/金字塔融合

引用本文复制引用

基金项目

国家重点研发计划项目(2018YFB1403303)

出版年

2023
电子测量技术
北京无线电技术研究所

电子测量技术

CSTPCD北大核心
影响因子：1.166
ISSN：1002-7300
被引量1
参考文献量5
段落导航相关论文