Pixel-Wise Color Constancy via Smoothness Techniques in Multi-Illuminant Scenes
Umut Cem Entok, Firas Laakom, Farhad Pakdaman, Moncef Gabbouj
TL;DR
This work tackles color constancy under scenes with multiple, spatially varying illuminants by predicting per-pixel illumination maps. It combines a U-Net-based estimator operating in the log-chrominance space with Total Variation loss to impose spatial smoothness, and uses a bilateral filter for photorealistic post-processing. To handle noisy ground-truth illumination, the authors apply label smoothing to the per-pixel weights. On the Large Scale Multi-Illuminant (LSMI) dataset, their PWCC methods achieve state-of-the-art recovery angular errors, with a mean around $2^\circ$, and show notable improvements on difficult cases; the approach offers practical improvements for robust white balancing in real-world, multi-illuminant scenes.
Abstract
Most scenes are illuminated by several light sources, where the traditional assumption of uniform illumination is invalid. This issue is ignored in most color constancy methods, primarily due to the complex spatial impact of multiple light sources on the image. Moreover, most existing multi-illuminant methods fail to preserve the smooth change of illumination, which stems from spatial dependencies in natural images. Motivated by this, we propose a novel multi-illuminant color constancy method, by learning pixel-wise illumination maps caused by multiple light sources. The proposed method enforces smoothness within neighboring pixels, by regularizing the training with the total variation loss. Moreover, a bilateral filter is provisioned further to enhance the natural appearance of the estimated images, while preserving the edges. Additionally, we propose a label-smoothing technique that enables the model to generalize well despite the uncertainties in ground truth. Quantitative and qualitative experiments demonstrate that the proposed method outperforms the state-of-the-art.
