液晶与显示2024,Vol.39Issue(8):1037-1045,9.DOI:10.37188/CJLCD.2023-0276
基于像素级水平的通道自适应水下图像增强算法
Channel adaptive underwater image enhancement algorithm based on pixel level
摘要
Abstract
The existing algorithms based on deep learning enhance underwater images in high-dimensional features by encoding and decoding,without considering the channel difference degradation characteristics of underwater images,resulting in generally poor enhancement effects.To solve this problem,this paper proposes an underwater image pixel-level channel enhancement algorithm based on deep learning,which enhances the underwater image at the pixel level into three channels:R,G and B.The algorithm is divided into four stages,and the whole enhancement process is completed through four stages of sub-channel feature extraction.The network first fixes the color channel of the context by enhancing the local and global semantics of the network and optimizing the channel attenuation.Secondly,the spatial and channel features are aggregated by an attention mechanism,and irrelevant color localization jump information is suppressed.Then,the adaptive features are adjusted by optimizing the attention mechanism.Finally,in order to improve the ability of color shift correction,a color shift correction module is proposed.In the fourth stage,a color shift adjustment module is used to further adjust the color shift problem of the image.Experimental results show that compared with other algorithms on the UIEB dataset and EUVP dataset,the proposed algorithm improves the PSNR index by 14.35%,the SSIM index by 5.8%,the UIQM index by 3.2%,and the UCIQE index by 13.7%,and has the best subjective effect.关键词
水下图像增强/通道增强/像素级增强/深度学习Key words
underwater image enhancement/channel enhancement/pixel-wise enhancement/deep learning分类
信息技术与安全科学引用本文复制引用
彭晏飞,张添淇,安彤..基于像素级水平的通道自适应水下图像增强算法[J].液晶与显示,2024,39(8):1037-1045,9.基金项目
国家自然科学基金(No.61772249) (No.61772249)
辽宁省高等学校基本科研项目(No.LJKZ0358)Supported by National Natural Science Foundation of China(No.61772249) (No.LJKZ0358)
Basic Scientific Research Project of Colleges and Universities of Liaoning Province(No.LJKZ0358) (No.LJKZ0358)