A General Pipeline for Dense Illuminant Estimation via Physically Based Synthetic Data
Organizations: Department of Informatics, Systems and Communication, University of Milano – Bicocca, Milan 20126, Italy
Abstract
Illuminant estimation is a fundamental problem in computational photography, as it enables the correction of color shifts induced by varying lighting conditions. While learning-based methods have demonstrated strong performance, their progress is hindered by the limited availability of large-scale datasets with accurate illuminant ground-truth. In this work, we propose a general and reusable pipeline to derive dense illuminant chromaticity maps from physically based 3D-rendered scenes. By repurposing an existing 3D scene collection, our approach enables the systematic generation of pixel-wise illuminant annotations under controlled lighting conditions, effectively lowering the barrier to data acquisition for learning-based illuminant estimation. Using this pipeline, we generate a large-scale synthetic set of 74,321 images, which we employ for pre-training both single- and multi-illuminant estimation models. Extensive experiments with state-of-the-art architectures show that synthetic pre-training consistently improves performance, with gains of up to 28% for single-illuminant estimation and up to 57% for multi-illuminant estimation, particularly in data-scarce regimes. These findings demonstrate that synthetic data generation pipelines offer an effective and scalable solution for the pre-training of illuminant estimation methods.
Figures & tables
| Hypersim-WB | LSMI-Galaxy [ 13 ] | MIMO [ 2 ] | |||||||||||
| Method | Pre-training Data | Mean | B-25 | W-25 | 95-P | Mean | B-25 | W-25 | 95-P | Mean | B-25 | W-25 | 95-P |
| GW [ 22 ] | - | 5.45 | 1.88 | 11.10 | 13.2 | 4.18 | 0.95 | 8.47 | 10.30 | 2.33 | 0.89 | 4.20 | 4.83 |
| UNet [ 13 ] | - | 3.18 | 1.1 | 6.35 | 8.00 | 1.94 | 0.65 | 3.88 | 4.63 | 3.30 | 2.06 | 5.48 | 6.08 |
| Hypersim-WB | - | - | - | - | 1.83 | 0.61 | 3.65 | 4.06 | 2.13 | 0.83 | 4.06 | 4.71 | |
| HDRNet [ 9 ] | - | 3.77 | 1.45 | 7.57 | 9.20 | 2.36 | 0.67 | 4.98 | 5.75 | 3.41 | 2.31 | 4.87 | 5.34 |
| Hypersim-WB | - | - | - | - | 2.19 | 0.59 | 4.65 | 5.68 | 3.16 | 1.71 | 5.19 | 5.73 | |
| ColorChecker [ 11 ] | |||||
| Method | Pre-training Data | Mean | B-25 | W-25 | 95-P |
| GW [ 22 ] | - | 4.80 | 1.72 | 7.11 | 13.20 |
| UNet [ 13 ] | - | 3.14 | 0.69 | 6.72 | 8.84 |
| Hypersim-WB | 2.94 | 0.63 | 6.76 | 7.55 | |
| HDRNet [ 9 ] | - | 3.53 | 0.90 | 7.20 | 9.25 |
| Hypersim-WB | 3.40 | 0.82 | 7.46 | 9.50 | |