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

你只需要一种色彩空间:一种用于低光照图像增强的高效网络

You Only Need One Color Space: An Efficient Network for Low-light Image Enhancement

School of Computer Science, Northwestern Polytechnical University(西北工业大学计算机学院);School of Computer Science, Xi’an University of Architecture and Technology(西安建筑科技大学计算机学院)

Qingsen Yan, Yixu Feng, Cheng Zhang, Pei Wang, Peng Wu, Wei Dong, Jinqiu Sun, Yanning Zhang

AI总结:

针对低光照增强中 sRGB/HSV 易放大噪声并产生伪影的问题,提出可训练 HVI 色彩空间与双分支 CIDNet,通过 LCA 交互解耦信息并降噪,在 11 个数据集上取得最优表现。

摘要:

低光照图像增强(LLIE)任务旨在从受损的低光照图像中恢复细节和视觉信息。大多数现有方法通过深度神经网络(DNN)在 sRGB 和 HSV 色彩空间上学习低光照/正常光照图像之间的映射函数。然而,增强过程涉及放大图像信号,而将这些色彩空间应用于具有低信噪比的低光照图像,会给增强过程带来敏感性和不稳定性。因此,这会导致增强后的图像中出现色彩伪影和亮度伪影。为缓解该问题,我们提出了一种新颖的可训练色彩空间,名为 Horizontal/Vertical-Intensity(HVI)。它不仅将亮度与色彩从 RGB 通道中解耦,以减轻增强过程中的不稳定性,而且由于具有可训练参数,还能够适应不同光照范围的低光照图像。此外,我们设计了一种新颖的色彩与强度解耦网络(Color and Intensity Decoupling Network,CIDNet),它包含两个分支,专门用于在 HVI 空间中处理被解耦的图像亮度和色彩。在 CIDNet 中,我们引入了轻量级交叉注意力(Lightweight Cross-Attention,LCA)模块,以促进两个分支中图像结构与内容信息之间的交互,同时抑制低光照图像中的噪声。最后,我们开展了 22 项定量和定性实验,结果表明所提出的 CIDNet 在 11 个数据集上优于最先进方法。代码可在 https://github.com/Fediory/HVI-CIDNet 获取。

英文摘要:

Low-Light Image Enhancement (LLIE) task tends to restore the details and visual information from corrupted low-light images. Most existing methods learn the mapping function between low/normal-light images by Deep Neural Networks (DNNs) on sRGB and HSV color space. Nevertheless, enhancement involves amplifying image signals, and applying these color spaces to low-light images with a low signal-to-noise ratio can introduce sensitivity and instability into the enhancement process. Consequently, this results in the presence of color artifacts and brightness artifacts in the enhanced images. To alleviate this problem, we propose a novel trainable color space, named Horizontal/Vertical-Intensity (HVI). It not only decouples brightness and color from RGB channels to mitigate the instability during enhancement but also adapts to low-light images in different illumination ranges due to the trainable parameters. Further, we design a novel Color and Intensity Decoupling Network (CIDNet) with two branches dedicated to processing the decoupled image brightness and color in the HVI space. Within CIDNet, we introduce the Lightweight Cross-Attention (LCA) module to facilitate interaction between image structure and content information in both branches, while also suppressing noise in low-light images. Finally, we conducted 22 quantitative and qualitative experiments to show that the proposed CIDNet outperforms the state-of-the-art methods on 11 datasets. The code is available at https://github.com/Fediory/HVI-CIDNet.

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

↑