红外技术2024，Vol.46Issue(9)：1006-1014,9.
基于注意力机制与特征重建的水下图像增强
Underwater Image Enhancement Based on Attention Mechanism and Feature Reconstruction
摘要
Abstract
To address the issues of existing underwater image enhancement methods,which lack focus on critical target objects in images and exhibit poor enhancement effects on edge detail information,in this study,an underwater image enhancement approach is proposed based on an attention mechanism and feature reconstruction.First,a superpixel image enhancement model is constructed by integrating the residual module with the Convolutional Block Attention Module(CBAM),which not only improves the overall quality of underwater images but also enhances the clarity and visibility of target objects in images.Second,an edge difference module is designed to enable the model to focus on high-frequency information in the images,thereby strengthening the edge details of the target objects.Finally,a multi-granularity feature reconstruction module is built to reconstruct the hidden layer features of the superpixel image enhancement model,restore the input image,and further optimize the model parameters.Experimental results demonstrate that when compared with contrastive methods,the proposed model realizes improvements in three evaluation metrics:Structural Similarity(SSIM),Peak Signal to Noise Ratio(PSNR),and Underwater Image Quality Measures(UIQM),indicating better enhancement performance.Notably,it exhibits a remarkable effect in enhancing critical target objects in underwater images.关键词
水下图像增强/超像素图像/注意力机制/边缘差分/多粒度特征重建Key words
underwater image enhancement/super-pixel image/attention mechanism/edge difference/multi-granularity feature reconstruction分类
计算机与自动化引用本文复制引用
王燕,张金峰,王丽康,范向辉..基于注意力机制与特征重建的水下图像增强[J].红外技术,2024,46(9):1006-1014,9.基金项目
国家自然科学基金(61863025). （61863025）
