基于红通道先验的双分支水下图像增强算法

    Dual-branch underwater image enhancement algorithm based on red channel prior

    • 摘要: 针对光在水中的吸收和散射作用引发的水下图像颜色失真、对比度低、细节模糊等问题,提出一种基于红通道先验的水下图像增强算法。将传统红通道先验理论与生成对抗网络相结合,设计了红通道先验模块用于关注水下图像红色通道的衰减特性并与生成器的编码器进行联合训练,使得编码器能够利用红通道先验模块提供的信息更好地提取有用特征,有利于解决水下图像色偏问题。判别器采用全局-局部双判别器,并构建了多个损失函数,使生成图像在结构、内容、色彩上和参考图像保持一致。在UIEB、EUVP、LSUI 3个公开数据集上与其他10种算法相比均取得了最优或次优的结果,在UIEB数据集上增强后的水下图像峰值信噪比(PSNR)、结构相似性(SSIM)、水下图像质量度量(UIQM)、水下彩色图像质量评估(UCIQE)、信息熵(IE)分别提高了1.78 dB、0.073、0.016、0.004和0.02,均方误差(MSE)下降了48.63。实验结果表明,本文所提算法在提升水下图像质量方面取得了显著成效,无论在主观视觉感受还是客观数据上都表现出明显优势。

       

      Abstract: Aiming at the problems of color distortion, low contrast and detail blur caused by light absorption and scattering in water, an underwater image enhancement algorithm based on red channel prior was proposed. Combining the traditional red channel prior theory with the generation adversarial network, a red channel prior module was designed to pay attention to the attenuation characteristics of the red channel in underwater images and conduct joint training with the encoder of the generator, so that the encoder could use the information provided by the red channel prior module to extract useful features better, which was conducive to solving the color bias problem of underwater images. The discriminator adopted global-local double discriminator, and constructed several loss functions to make the generated image consistent with the reference image in structure, content and color. Compared with other 10 algorithms, the UIEB, EUVP and LSUI public data sets obtained optimal or sub-optimal results. The enhanced underwater image peak signal-to-noise ratio (PSNR), structural similarity (SSIM), underwater image quality metric (UIQM), underwater color image quality assessment (UCIQE) and information entropy(IE) on the UIEB dataset were increased by 1.78 dB, 0.073, 0.016, 0.004 and 0.02, respectively. Mean square error (MSE) decreased by 48.63. The algorithm proposed in this paper has achieved remarkable results in improving the quality of underwater images, showing obvious advantages in both subjective visual perception and objective data.

       

    /

    返回文章
    返回