基于非局部Retinex的图像增强方法,无需深度学习也能超越多数模型。
Nonlocal Retinex-Based Variational Model and its Deep Unfolding Twin for Low-Light Image Enhancement
- 将光照、反射率与噪声分离,用非局部梯度项保留结构细节。
- 在多个数据集上优于主流方法,即使不依赖学习也能达到顶尖性能。
- 适合需要高可解释性或轻量级部署的低光图像增强场景。
低光条件下拍摄的图像在诸多应用中存在显著缺陷,如细节模糊、对比度下降和噪声隐藏。去除光照影响并提升图像质量对图像分割、目标检测等任务至关重要。本文提出一种基于Retinex分解的变分图像增强方法,将图像分解为光照、反射率和噪声三部分,并引入色彩校正预处理步骤作为分解输入。模型设计了新型非局部梯度型保真项以保留结构细节,并加入自动伽马校正模块。进一步,构建了该变分模型的深度展开版本,将近端算子替换为可学习网络,并引入交叉注意力机制,捕捉反射率的非局部先验与非局部梯度约束中的长程依赖。实验结果表明,两种方法在多个数据集上均优于近期及前沿技术。尤其值得注意的是,尽管不依赖学习策略,该变分模型在视觉效果和质量指标上仍超过多数深度学习方法。
原文摘要 · Abstract (English)
Images captured under low-light conditions present significant limitations in many applications, as poor lighting can obscure details, reduce contrast, and hide noise. Removing the illumination effects and enhancing the quality of such images is crucial for many tasks, such as image segmentation and object detection. In this paper, we propose a variational method for low-light image enhancement based on the Retinex decomposition into illumination, reflectance, and noise components. A color correction pre-processing step is applied to the low-light image, which is then used as the observed input in the decomposition. Moreover, our model integrates a novel nonlocal gradient-type fidelity term designed to preserve structural details. Additionally, we propose an automatic gamma correction module. Building on the proposed variational approach, we extend the model by introducing its deep unfolding counterpart, in which the proximal operators are replaced with learnable networks. We propose cross-attention mechanisms to capture long-range dependencies in both the nonlocal prior of the reflectance and the nonlocal gradient-based constraint. Experimental results demonstrate that both methods compare favorably with several recent and state-of-the-art techniques across different datasets. In particular, despite not relying on learning strategies, the variational model outperforms most deep learning approaches both visually and in terms of quality metrics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。