论文标题
基于注意力的低光图像增强网络
Attention-based network for low-light image enhancement
论文作者
论文摘要
在弱光条件下捕获的图像通常会遭受亮度和臭名昭著的噪音。因此,低光图像增强是计算机视觉中的关键挑战性任务。已经提出了针对此任务的多种方法,但是这些方法通常在极端的低光环境中失败,并放大了输入图像中的基本噪声。为了解决这一困难问题,本文提出了一个新型的基于注意力的神经网络,以从原始传感器数据中产生高质量增强的低光图像。具体而言,我们首先采用注意力策略(即通道注意力和空间注意模块)来抑制不希望的色差和噪声。通道注意模块指导网络提高冗余颜色功能。空间注意模块通过利用图像中的非局部相关性来重点介绍denoing。此外,我们提出了一个新的合并层,称为倒流层,该层可自适应地从先前的功能中选择有用的信息。广泛的实验表明,在增强中抑制色差和噪声伪像的角度,尤其是当弱光图像具有严重的噪声时,提出的网络的优越性。
The captured images under low light conditions often suffer insufficient brightness and notorious noise. Hence, low-light image enhancement is a key challenging task in computer vision. A variety of methods have been proposed for this task, but these methods often failed in an extreme low-light environment and amplified the underlying noise in the input image. To address such a difficult problem, this paper presents a novel attention-based neural network to generate high-quality enhanced low-light images from the raw sensor data. Specifically, we first employ attention strategy (i.e. channel attention and spatial attention modules) to suppress undesired chromatic aberration and noise. The channel attention module guides the network to refine redundant colour features. The spatial attention module focuses on denoising by taking advantage of the non-local correlation in the image. Furthermore, we propose a new pooling layer, called inverted shuffle layer, which adaptively selects useful information from previous features. Extensive experiments demonstrate the superiority of the proposed network in terms of suppressing the chromatic aberration and noise artifacts in enhancement, especially when the low-light image has severe noise.