White balance (WB) is a key step in the image signal processor (ISP) pipeline that mitigates color casts caused by varying illumination and restores the scene's true colors. Currently, sRGB-based WB editing for post-ISP WB correction is widely used to address color constancy failures in the ISP pipeline when the original camera RAW is unavailable. However, additive color models (e.g., sRGB) are inherently limited by fixed nonlinear transformations and entangled color channels, which often impede their generalization to complex lighting conditions. To address these challenges, we propose a novel framework for WB correction that leverages a perception-inspired Learnable HSI (LHSI) color space. Built upon a cylindrical color model that naturally separates luminance from chromatic components, our framework further introduces dedicated parameters to enhance this disentanglement and learnable mapping to adaptively refine the flexibility. Moreover, a new Mamba-based network is introduced, which is tailored to the characteristics of the proposed LHSI color space. Experimental results on benchmark datasets demonstrate the superiority of our method, highlighting the potential of perception-inspired color space design in computational photography. The source code is available at https://github.com/YangCheng58/WB_Color_Space.


翻译:白平衡(WB)是图像信号处理器(ISP)流水线中的关键步骤,用于减轻由不同光照引起的色偏并恢复场景的真实色彩。目前,基于sRGB的后ISP白平衡编辑被广泛用于解决当原始相机RAW数据不可用时ISP流水线中的色彩恒常性失效问题。然而,加色模型(如sRGB)固有地受限于固定的非线性变换和纠缠的色彩通道,这往往阻碍了其向复杂光照条件的泛化。为应对这些挑战,我们提出了一种新颖的白平衡校正框架,该框架利用了一种感知启发的可学习HSI(LHSI)色彩空间。该框架基于一种自然分离亮度与色度分量的柱面色彩模型构建,进一步引入了专用参数以增强这种解耦,并通过可学习的映射自适应地提升灵活性。此外,我们引入了一种新的基于Mamba的网络,该网络专为所提出的LHSI色彩空间的特性而设计。在基准数据集上的实验结果表明了我们方法的优越性,凸显了感知启发性色彩空间设计在计算摄影学中的潜力。源代码可在 https://github.com/YangCheng58/WB_Color_Space 获取。

0
下载
关闭预览

相关内容

设计是对现有状的一种重新认识和打破重组的过程,设计让一切变得更美。
【CVPR2025】基于组合表示移植的图像编辑方法
专知会员服务
8+阅读 · 4月5日
NAACL 2019 | 一种考虑缓和KL消失的简单VAE训练方法
PaperWeekly
20+阅读 · 2019年4月24日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
相关基金
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员