基于两阶段注意力机制的快速水下图像增强方法(英文) |
| |
引用本文: | 蔚婧,张乐,吴萌,江尚航,李道江.基于两阶段注意力机制的快速水下图像增强方法(英文)[J].东南大学学报,2023(4):410-415. |
| |
作者姓名: | 蔚婧 张乐 吴萌 江尚航 李道江 |
| |
作者单位: | 西北工业大学航海学院 |
| |
基金项目: | The Natural Science Basic Research Plan in Shaanxi Province of China (No.2020JQ-208);;Key Research and Development Program of Shaanxi (No.2022GY-285, No.2020SF-391); |
| |
摘 要: | 为提升水下退化图像的质量,提出了一种基于两阶段注意力机制的快速水下图像增强算法.首先,在特征内采用自注意力机制来加强网络对重要信息的关注.然后,通过交叉注意力机制,将基于物理先验的水下传输图融入网络,以增强网络对质量退化区域的特征表达.最后,从主观和客观标准出发,设计了多项联合损失函数,引导网络获得更好的视觉增强效果.3个基准数据集上的实验结果表明,与5种水下图像增强方法相比,所提方法的峰值信噪比和结构相似性分数更高,性能明显更优.该方法不仅能有效恢复图像颜色和纹理细节,而且具备实时高效的处理速度.
|
关 键 词: | 水下图像增强 自注意力 交叉注意力 传输图 |
|
|