arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.06705cs.CV

Retinexformer:基于Retinex的单阶段Transformer用于低光图像增强

Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement

Tsinghua University(清华大学);University of Würzburg(维尔茨堡大学);ETH Zürich(苏黎世联邦理工学院)

Yuanhao Cai, Hao Bian, Jing Lin, Haoqian Wang, Radu Timofte, Yulun Zhang

AI总结:

本文提出单阶段Retinex框架ORF和光照引导Transformer(IGT),构建Retinexformer,在十三个基准上显著超越现有低光图像增强方法。

摘要:

在增强低光图像时,许多深度学习算法基于Retinex理论。然而,Retinex模型并未考虑隐藏在暗处或由提亮过程引入的损坏。此外,这些方法通常需要繁琐的多阶段训练流程,并且依赖卷积神经网络,在捕获长程依赖方面存在局限性。本文中,我们构建了一个简单但有原则的单阶段Retinex框架(ORF)。ORF首先估计光照信息以提亮低光图像,然后恢复损坏以生成增强图像。我们设计了一个光照引导Transformer(IGT),利用光照表示来指导对不同光照条件区域之间非局部交互的建模。通过将IGT嵌入ORF,我们得到了我们的算法Retinexformer。全面的定量和定性实验表明,我们的Retinexformer在十三个基准上显著优于最先进的方法。用户研究和在低光目标检测上的应用也揭示了我们方法潜在的实际价值。代码、模型和结果可在 https://github.com/caiyuanhao1998/Retinexformer 获取。

英文摘要:

When enhancing low-light images, many deep learning algorithms are based on the Retinex theory. However, the Retinex model does not consider the corruptions hidden in the dark or introduced by the light-up process. Besides, these methods usually require a tedious multi-stage training pipeline and rely on convolutional neural networks, showing limitations in capturing long-range dependencies. In this paper, we formulate a simple yet principled One-stage Retinex-based Framework (ORF). ORF first estimates the illumination information to light up the low-light image and then restores the corruption to produce the enhanced image. We design an Illumination-Guided Transformer (IGT) that utilizes illumination representations to direct the modeling of non-local interactions of regions with different lighting conditions. By plugging IGT into ORF, we obtain our algorithm, Retinexformer. Comprehensive quantitative and qualitative experiments demonstrate that our Retinexformer significantly outperforms state-of-the-art methods on thirteen benchmarks. The user study and application on low-light object detection also reveal the latent practical values of our method. Code, models, and results are available at https://github.com/caiyuanhao1998/Retinexformer

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑