让水下图像可调色,一次处理生成多种真实色彩效果。
Underwater Organism Color Enhancement via Color Code Decomposition, Adaptation and Interpolation
- 将颜色与内容解耦,通过高斯分布采样实现可控调色。
- 在多个挑战性数据集上优于现有方法,色彩更真实多样。
- 适合需要灵活调整水下图像色彩的科研与应用人员。
水下图像常因光吸收和散射导致质量下降。现有增强算法多生成单一固定色彩图像,限制了用户灵活性和应用范围。为此,我们提出名为 ColorCode 的方法,在提升水下图像质量的同时,支持多种可控色彩输出。该方法通过监督训练恢复出参考增强图像,并利用自重构与交叉重构将其分解为颜色码和内容码。颜色码被显式约束为符合高斯分布,从而在推理阶段实现高效采样与插值。ColorCode 具备三大特性:1)颜色增强,生成固定色彩的优化图像;2)颜色适配,通过引导图像控制长波段颜色成分;3)颜色插值,通过连续采样颜色码平滑生成多种色彩。在多个流行且具有挑战性的基准数据集上的定量与视觉评估表明,ColorCode 在提供多样化、可调控且色彩真实的增强结果方面优于现有方法。源代码已公开于 https://github.com/Xiaofeng-life/ColorCode。
原文摘要 · Abstract (English)
Underwater images often suffer from quality degradation due to absorption and scattering effects. Most existing underwater image enhancement algorithms produce a single, fixed-color image, limiting user flexibility and application. To address this limitation, we propose a method called \textit{ColorCode}, which enhances underwater images while offering a range of controllable color outputs. Our approach involves recovering an underwater image to a reference enhanced image through supervised training and decomposing it into color and content codes via self-reconstruction and cross-reconstruction. The color code is explicitly constrained to follow a Gaussian distribution, allowing for efficient sampling and interpolation during inference. ColorCode offers three key features: 1) color enhancement, producing an enhanced image with a fixed color; 2) color adaptation, enabling controllable adjustments of long-wavelength color components using guidance images; and 3) color interpolation, allowing for the smooth generation of multiple colors through continuous sampling of the color code. Quantitative and visual evaluations on popular and challenging benchmark datasets demonstrate the superiority of ColorCode over existing methods in providing diverse, controllable, and color-realistic enhancement results. The source code is available at https://github.com/Xiaofeng-life/ColorCode.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。