文献类型：期刊文献

英文题名：SCAUIE-Net: Underwater Image Enhancement Method Based on Spatial and Channel Attention

作者：Zhong, Yuanhao[1];Wang, Ji[1,2];Lu, Qingjie[1]

机构：[1]Guangdong Ocean Univ, Sch Elect & Informat Engn, Zhanjiang 524088, Guangdong, Peoples R China;[2]Guangdong Smart Ocean Sensor Network & Its Equipme, Zhanjiang 524088, Guangdong, Peoples R China

年份：2023

卷号：11

起止页码：72172

外文期刊名：IEEE ACCESS

收录：SCI-EXPANDED(收录号：WOS:001038315500001)、、EI(收录号：20232814385449)、Scopus(收录号：2-s2.0-85164441376)、WOS

语种：英文

外文关键词：Underwater image enhancement; low-level vision; attention mechanism

外文摘要：Underwater image enhancement is a Low-Level Vision task that plays an important role in marine resource development, but the light absorption and scattering cause severe underwater image quality degradation. To solve these problems, this paper proposes a neural network based on a spatial and channel attention module that reinforces the network's attention to channel and spatial information. The network's Confidence Generator can precisely extract feature maps from multi-scale underwater images. Meanwhile, we propose a new training loss function by mixing perceptual, MS-SSIM and MAE loss functions to further improve the contrast in high-frequency, colors and luminance. For training, this paper also uses a feature fusion strategy: Firstly, augmenting the training underwater images by Gamma Correction, White Balance and Histogram Equalization algorithms to remove color cast, lighten up dark regions and improve the contrast. Then, fusing the enhancing images with confidence maps predicted from the Generator. The network was validated in the UIEB dataset and obtains efficient improvements on Peak Signal-to-Noise Ratio (PSNR) and Structural Similarity Index (SSIM) metrics, yielding a PSNR of 22.9286 and SSIM of 0.9290. Experimental results on real-world underwater images demonstrate that the proposed method performs well on different underwater scenes.