首页 | 本学科首页   官方微博 | 高级检索  
     检索      

基于两阶段注意力机制的快速水下图像增强方法(英文)
引用本文:蔚婧,张乐,吴萌,江尚航,李道江.基于两阶段注意力机制的快速水下图像增强方法(英文)[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种水下图像增强方法相比,所提方法的峰值信噪比和结构相似性分数更高,性能明显更优.该方法不仅能有效恢复图像颜色和纹理细节,而且具备实时高效的处理速度.

关 键 词:水下图像增强  自注意力  交叉注意力  传输图
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号