arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.04366cs.CV

用于图像曝光校正的CNN注入Transformer

CNN Injected Transformer for Image Exposure Correction

University of Macau(澳门大学)

Shuning Xu, Xiangyu Chen, Binbin Song, Jiantao Zhou

AI总结:

针对卷积方法因感受野受限导致曝光偏差的问题,提出CNN注入Transformer(CIT),结合两者优势并引入辅助模块与损失函数,实现优于现有方法的图像曝光校正。

摘要:

使用不正确的曝光设置捕获图像会导致视觉体验不佳。只有正确设置曝光,才能适当地保留图像的颜色和细节。以往基于卷积的曝光校正方法由于卷积核感受野受限,常导致图像出现曝光偏差。这是因为卷积无法准确捕捉图像中的长距离依赖关系。为克服这一挑战,我们可以应用Transformer来解决曝光校正问题,利用其建模长距离依赖的能力来捕捉全局表示。然而,仅依赖基于窗口的Transformer会因在小图块上应用自注意力而导致视觉上令人不适的块状伪影。本文提出一种CNN注入Transformer(CIT),以同时发挥CNN和Transformer各自的优势。具体而言,我们利用基于窗口的Transformer构建CIT,以挖掘整幅图像中不同区域间的长距离交互。在每个CIT块内,我们引入通道注意力块(CAB)和半实例归一化块(HINB),辅助基于窗口的自注意力获取全局统计信息并细化局部特征。除了用于曝光校正的混合架构设计外,我们还应用一组精心设计的损失函数来改善空间相干性并纠正潜在的颜色偏差。大量实验表明,我们的图像曝光校正方法在定量和定性指标上均优于现有最先进的方法。

英文摘要:

Capturing images with incorrect exposure settings fails to deliver a satisfactory visual experience. Only when the exposure is properly set, can the color and details of the images be appropriately preserved. Previous exposure correction methods based on convolutions often produce exposure deviation in images as a consequence of the restricted receptive field of convolutional kernels. This issue arises because convolutions are not capable of capturing long-range dependencies in images accurately. To overcome this challenge, we can apply the Transformer to address the exposure correction problem, leveraging its capability in modeling long-range dependencies to capture global representation. However, solely relying on the window-based Transformer leads to visually disturbing blocking artifacts due to the application of self-attention in small patches. In this paper, we propose a CNN Injected Transformer (CIT) to harness the individual strengths of CNN and Transformer simultaneously. Specifically, we construct the CIT by utilizing a window-based Transformer to exploit the long-range interactions among different regions in the entire image. Within each CIT block, we incorporate a channel attention block (CAB) and a half-instance normalization block (HINB) to assist the window-based self-attention to acquire the global statistics and refine local features. In addition to the hybrid architecture design for exposure correction, we apply a set of carefully formulated loss functions to improve the spatial coherence and rectify potential color deviations. Extensive experiments demonstrate that our image exposure correction method outperforms state-of-the-art approaches in terms of both quantitative and qualitative metrics.

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

↑