arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.04740eess.IV
收藏

色彩校正遇上跨频谱细化:一种用于水下图像复原的分布感知扩散模型

Color Correction Meets Cross-Spectral Refinement: A Distribution-Aware Diffusion for Underwater Image Restoration

Laibin Chang, Yunke Wang, Bo Du, Chang Xu

AI总结:

针对水下图像增强中扩散模型计算开销高且存在色彩不平衡扰动的问题,本文提出DiffColor,结合小波域扩散、GCC色彩校正与CSDR细节细化,实现高效高保真复原并取得领先性能。

摘要:

水下成像通常会遭受严重的视觉退化,这限制了其对后续应用的适用性。尽管近期的水下图像增强(UIE)方法依赖于深度神经网络架构设计的最新进展,但在跨场景鲁棒性和计算效率方面仍有相当大的提升空间。扩散模型已在图像生成中取得巨大成功,这促使我们考虑将其应用于UIE任务。然而,将其直接应用于UIE任务会带来两个挑战,即高昂的计算开销和色彩不平衡扰动。为解决这些问题,我们提出了DiffColor,这是一种用于高效UIE的分布感知扩散与跨频谱细化模型。我们没有在原始像素空间中进行扩散,而是将图像转换到小波域以获得低频和高频频谱,该变换在每次转换后本质上将图像空间维度减半。与单噪声图像复原任务不同,由于水对光的选择性吸收,水下成像呈现出不平衡的通道分布。针对这一点,我们设计了全局色彩校正(GCC)模块来处理各种色彩偏移,从而避免去噪过程中潜在的全局退化干扰。对于水下散射造成的图像细节损失,我们进一步提出跨频谱细节细化(CSDR)来增强高频细节,并将其与低频信号整合,作为引导扩散的输入条件。这种方式不仅确保了采样内容的高保真度,还补偿了损失的细节。综合实验表明,DiffColor在定量和定性评估中均优于最先进方法。

英文摘要:

Underwater imaging often suffers from significant visual degradation, which limits its suitability for subsequent applications. While recent underwater image enhancement (UIE) methods rely on the current advances in deep neural network architecture designs, there is still considerable room for improvement in terms of cross-scene robustness and computational efficiency. Diffusion models have shown great success in image generation, prompting us to consider their application to UIE tasks. However, directly applying them to UIE tasks will pose two challenges, \textit{i.e.}, high computational budget and color unbalanced perturbations. To tackle these issues, we propose DiffColor, a distribution-aware diffusion and cross-spectral refinement model for efficient UIE. Instead of diffusing in the raw pixel space, we transfer the image into the wavelet domain to obtain such low-frequency and high-frequency spectra, it inherently reduces the image spatial dimensions by half after each transformation. Unlike single-noise image restoration tasks, underwater imaging exhibits unbalanced channel distributions due to the selective absorption of light by water. To address this, we design the Global Color Correction (GCC) module to handle the diverse color shifts, thereby avoiding potential global degradation disturbances during the denoising process. For the sacrificed image details caused by underwater scattering, we further present the Cross-Spectral Detail Refinement (CSDR) to enhance the high-frequency details, which are integrated with the low-frequency signal as input conditions for guiding the diffusion. This way not only ensures the high-fidelity of sampled content but also compensates for the sacrificed details. Comprehensive experiments demonstrate the superior performance of DiffColor over state-of-the-art methods in both quantitative and qualitative evaluations.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑