基于优化CycleGAN的水下图像增强算法

    Underwater image enhancement algorithm based on optimized CycleGAN

    • 摘要: 针对水下图像存在色偏、雾化严重和对比度低等问题,提出了一种基于改进CycleGAN (cycle-consistent generative adversarial networks)的水下图像增强算法。首先,设计了一种多尺度颜色校正模块,该模块可以通过捕获多层次的颜色和纹理信息来解决水下图像的色偏问题,并将其引入到CycleGAN的判别器网络中,使判别器可以更好地区分真实图像和生成图像。其次,在判别器中引入CBAM (convolutional block attention module)注意力模块,提高了模型对重要特征的关注度。此外,在U-Net生成器中加入多尺度特征提取模块来处理不同尺度的特征和细节,并使用SmoothL1损失函数替换CycleGAN网络中的MSE损失,增强了模型的泛化能力。最后,在UIEBD和EUVP数据集中进行实验验证,实验结果表明,本文算法与CycleGAN相比,UCIQE (underwater color image quality evaluation)平均值提高了10%,UIQM (underwater image quality measure)提高了25%,验证了本文算法能够满足水下图像增强的需求。

       

      Abstract: An underwater image enhancement algorithm was proposed based on an improved CycleGAN to address issues such as color distortion, severe fogging, and low contrast in underwater images. First, a multi-scale color correction module was designed to solve the color bias problem in underwater images by capturing multi-level color and texture information. This module was integrated into the discriminator network of CycleGAN, enabling the discriminator to better distinguish between real and generated images. Secondly, a CBAM (convolutional block attention module) attention module was inserted into the discriminator to enhance the model's focus on important features. Additionally, a multi-scale feature extraction module was added to the U-Net generator to handle features and details at different scales, and the SmoothL1 loss function was used to replace the MSE (mean square error) loss in the CycleGAN network, which could improve the model's generalization ability. Finally, experiments were conducted on the UIEBD and EUVP datasets, and the results show that, compared to the original algorithm, the proposed method could improve the average UCIQE(underwater color image quality evaluation) by 10% and the UIQM by 17%, demonstrating that the proposed algorithm meets the requirements for underwater image enhancement.

       

    /

    返回文章
    返回