一个模型搞定所有光照条件下的图像修复,通用性强。
OmniLight: One Model to Rule All Lighting Conditions

- 提出基于小波域专家混合的统一框架,跨数据集训练提升泛化能力。
- 在NTIRE 2026挑战赛中三项光照修复任务均获顶尖排名。
- 适合需要应对复杂真实光照场景的视觉系统开发者使用。
恶劣光照条件如投影阴影和不均匀照明会严重降低图像可见度与色彩保真度,影响计算机视觉系统的性能。因此,有效的阴影去除与自适应光照归一化(ALN)对恢复图像内容、提升感知质量及下游任务鲁棒性至关重要。针对现有方法多聚焦特定基准而缺乏跨域适应性的局限,本文对比研究了两种策略:以DINOLight为专用基线,利用各数据集特性;并提出通用模型OmniLight,融合所提的小波域专家混合(WD-MoE)结构,在全部提供数据集上联合训练。通过对比分析,揭示了数据分布对专用与统一架构性能的影响。两者在NTIRE 2026挑战赛的三项光照相关任务中均取得顶尖成绩,验证了其卓越的感知质量和泛化能力。代码已开源。
原文摘要 · Abstract (English)
Adverse lighting conditions, such as cast shadows and irregular illumination, pose significant challenges to computer vision systems by degrading visibility and color fidelity. Consequently, effective shadow removal and ALN are critical for restoring underlying image content, improving perceptual quality, and facilitating robust performance in downstream tasks. However, while achieving state-of-the-art results on specific benchmarks is a primary goal in image restoration challenges, real-world applications often demand robust models capable of handling diverse domains. To address this, we present a comprehensive study on lighting-related image restoration by exploring two contrasting strategies. We leverage a robust framework for ALN, DINOLight, as a specialized baseline to exploit the characteristics of each individual dataset, and extend it to OmniLight, a generalized alternative incorporating our proposed Wavelet Domain Mixture-of-Experts (WD-MoE) that is trained across all provided datasets. Through a comparative analysis of these two methods, we discuss the impact of data distribution on the performance of specialized and unified architectures in lighting-related image restoration. Notably, both approaches secured top-tier rankings across all three lighting-related tracks in the NTIRE 2026 Challenge, demonstrating their outstanding perceptual quality and generalization capabilities. Our codes are available at https://github.com/OBAKSA/Lighting-Restoration.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。