基于多尺度加权CycleGAN的水下图像增强算法

Underwater Image Enhancement Algorithm Based on Multi-Scale Weighting CycleGAN

  • 摘要: 针对水下图像色偏严重、雾化模糊和细节信息丢失等问题,提出一种基于多尺度加权CycleGAN的水下图像增强算法。首先,设计一种基于注意力机制的多尺度加权模块,并将其嵌入CycleGAN的残差结构,使网络动态地调整对水下图像不同区域的关注度,以校正水下图像的色偏与颜色失真;然后,设计一种跨尺度交互注意力模块将不同尺度下的特征整合,实现跨尺度的注意力加权,丰富增强图像的局部细节和全局结构信息。最后,引入结构相似性(Structure Similarity Index Measure,SSIM)损失函数对生成器训练过程进行监督,进一步提高生成图像的质量。实验结果表明,在测试集上进行水下彩色图像质量评价指标(Underwater Color Image Quality Evaluation,UCIQE)、水下图像质量指标(Underwater Image Quality Measures,UIQM)和信息熵的量化分析结果平均值分别为0.6022、4.426、7.562,相比较原始图像平均值提高了17.2%、32.7%和3.2%。所提算法在增强水下图像质量的同时,保留了更丰富的信息。

     

    Abstract: To address issues such as severe color cast, haziness, and loss of detail in underwater images, this paper proposes an underwater image enhancement algorithm based on a multi-scale weighted CycleGAN. First, a multi-scale weighted module based on a self-attention mechanism is proposed and embedded into the residual structure of CycleGAN, allowing the network to dynamically adjust its focus on different regions of the underwater image to correct color cast and color distortion. Then, multi-scale feature fusion is employed to integrate the weighted features, enriching the local details and global structural information in the enhanced image. Finally, the Structural Similarity Index Measure (SSIM) loss function is incorporated to supervise the training process of the generator, further improving the quality of the generated images. Experimental results show that, on the test set, the proposed method achieves average values of 0.6022, 4.426, and 7.562 for the Underwater Color Image Quality Evaluation (UCIQE), Underwater Image Quality Measures (UIQM), and information entropy, respectively. These represent improvements of 17.2%, 32.7%, and 3.2% over the original images. The proposed algorithm not only enhances the quality of underwater images but also preserves more detailed information.

     

/

返回文章
返回