液晶与显示2025,Vol.40Issue(8):1163-1176,14.DOI:10.37188/CJLCD.2025-0072
基于颜色先验引导和注意力机制的水下图像增强
Underwater image enhancement based on color prior guidance and attention mechanism
摘要
Abstract
To address the degradation issues in underwater images,such as color distortion,contrast reduction,and detail blurring,we propose an underwater image enhancement algorithm based on color prior guidance and attention mechanisms.We designed a Bayesian posterior feature extraction module to fuse color-prior-corrected image with the original image,and a color prior guided attention Transformer to utilize the prior information to guide the image enhancement process,which reduces the burden of color restoration during training.We constructed a hybrid multi-scale attention module to enhance feature representation in key regions.A dual-axis decoupled attention module was introduced to eliminate feature redundancy at the bottleneck layer,suppress overfitting,and improve the detail restoration.The proposed algorithm achieves PSNR/SSIM scores of 24.33/0.910 9 on UIEB,28.40/0.885 9 on UFO,and 29.00/0.899 1 on EUVP,demonstrating superior performance compared to state-of-the-art approaches on all three benchmarks.In addition,the proposed algorithm improves the UIQM and UCIQE metrics by 12.81%and 5.19%,respectively,compared to the original images.These results verify the effectiveness of the proposed algorithm in improving image sharpness,structural similarity,and overall visual quality.关键词
水下图像增强/颜色先验引导/注意力机制Key words
underwater image enhancement/color prior guidance/attention mechanism分类
信息技术与安全科学引用本文复制引用
李文彪,陶洋,董源,周立群..基于颜色先验引导和注意力机制的水下图像增强[J].液晶与显示,2025,40(8):1163-1176,14.基金项目
2023年汽车核心软件研发重大专项(No.CSTB2023TIAD-STX0027) (No.CSTB2023TIAD-STX0027)
国家重点研发计划(No.2019YFB2102001)Supported by 2023 Automotive Core Software Research and Development Major Project(No.CSTB2023TIAD-STX0027) (No.2019YFB2102001)
National Key R&D Program of China(No.2019YFB2102001) (No.2019YFB2102001)