高级检索

面向水下图像增强的内容与风格双域改善方法

Content and Style Dual-Domain Improvement Method for Underwater Image Enhancement

  • 摘要: 水下图像增强技术对于海洋资源开发具有重要意义。针对现有的水下图像增强方法难以应对现实世界复杂的水下环境,尤其对于困难样本上增强效果不佳的问题,提出一种基于内容与风格双域改善的水下图像增强方法。首先设计一个简单有效的框架,将水下图像解耦为内容特征与风格特征并重建;对内容域设计了双域自适应高频调整模块(dual-domain adaptive high-frequency adjust block,DAHAB),增强原始水下图像内容特征中的细节并抑制噪声;对风格域建立风格库,并在推理过程中匹配最优风格实现风格的变换;最后通过构建内容一致性损失和重建损失,监督模型完成水下图像的重建与增强。实验结果表明,在水下图像增强领域公开的UIEB、EUVP和RUIE数据集上,所提方法的主观视觉效果和客观评价指标(PSNR、SSIM、UIQM、UCIQE和MUSIQ)均优于对比方法,尤其在增强困难样本方面展现了明显的优势。

     

    Abstract: Underwater image enhancement techniques are crucial for the development of marine resources. To address the difficulties faced by existing underwater image enhancement methods in dealing with the complex underwater environments of the real world, especially in effectively enhancing challenging samples, we propose a method based on dual-domain improvement in content and style. Firstly, a simple and effective framework is designed to decouple underwater images into content features and style features, followed by reconstruction. For the content domain, a dual-domain adaptive high-frequency adjustment block is designed to enhance detail and suppress noise in the content features of the original underwater images. For the style domain, a style library is established to enable style transformation by matching the optimal style during inference. Finally, the content consistency loss and the reconstruction loss are used to supervise the model in completing the reconstruction and enhancement of underwater images. Experimental results on the publicly available underwater image datasets UIEB, EUVP, and RUIE show that our method outperforms compared methods in both subjective visual effects and objective evaluation metrics (PSNR, SSIM, UIQM, UCIQE, and MUSIQ). Notably, it demonstrates significant advantages in enhancing challenging samples.

     

/

返回文章
返回