arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.05906eess.IVcs.CV

面向屏下相机图像复原的分割引导稀疏Transformer

Segmentation Guided Sparse Transformer for Under-Display Camera Image Restoration

Shenzhen Campus of Sun Yat-sen University(中山大学深圳校区);Hong Kong University of Science and Technology(香港科技大学);State Key Laboratory for Novel Software Technology, Nanjing University(南京大学计算机软件新技术国家重点实验室);Samsung Research China - Beijing (SRC-B)(三星中国研究院北京分部);Harbin Institute of Technology, Shenzhen(哈尔滨工业大学深圳校区);SAMSUNG Electronics(三星电子)

Jingyun Xue, Tao Wang, Pengwen Dai, Kaihao Zhang

AI总结:

针对屏下相机图像退化问题,提出分割引导的稀疏Transformer方法SGSFormer,以实例分割为先验引导稀疏自注意力过滤冗余噪声,聚焦退化区域特征,提升复原质量。

摘要:

屏下相机(Under-Display Camera,UDC)是一种通过将摄像头隐藏在显示面板下方实现全屏显示的新兴技术。但目前的UDC实现方案会引发严重的图像退化问题:相机成像所需的入射光穿过显示面板时会发生衰减与衍射,导致UDC成像出现多种伪影。当前主流的UDC图像复原方法大多采用卷积神经网络架构,而基于Transformer的方法已在多数图像复原任务中展现出更优性能,这得益于Transformer能够采样全局特征以实现图像的局部重建,从而达成高质量的图像复原。本文研究发现,使用视觉Transformer(Vision Transformer)进行UDC退化图像复原时,全局注意力会采样大量冗余信息与噪声。此外,与采用密集注意力的普通Transformer相比,使用稀疏注意力的Transformer能够缓解冗余信息与噪声带来的不利影响。基于这一发现,我们提出了一种分割引导稀疏Transformer方法(SGSFormer),用于从UDC退化图像中复原高质量图像。具体而言,我们利用稀疏自注意力过滤冗余信息与噪声,引导模型将注意力集中在与待重建退化区域更相关的特征上。此外,我们将实例分割图作为先验信息整合进来,引导稀疏自注意力完成过滤并聚焦于正确区域。

英文摘要:

Under-Display Camera (UDC) is an emerging technology that achieves full-screen display via hiding the camera under the display panel. However, the current implementation of UDC causes serious degradation. The incident light required for camera imaging undergoes attenuation and diffraction when passing through the display panel, leading to various artifacts in UDC imaging. Presently, the prevailing UDC image restoration methods predominantly utilize convolutional neural network architectures, whereas Transformer-based methods have exhibited superior performance in the majority of image restoration tasks. This is attributed to the Transformer's capability to sample global features for the local reconstruction of images, thereby achieving high-quality image restoration. In this paper, we observe that when using the Vision Transformer for UDC degraded image restoration, the global attention samples a large amount of redundant information and noise. Furthermore, compared to the ordinary Transformer employing dense attention, the Transformer utilizing sparse attention can alleviate the adverse impact of redundant information and noise. Building upon this discovery, we propose a Segmentation Guided Sparse Transformer method (SGSFormer) for the task of restoring high-quality images from UDC degraded images. Specifically, we utilize sparse self-attention to filter out redundant information and noise, directing the model's attention to focus on the features more relevant to the degraded regions in need of reconstruction. Moreover, we integrate the instance segmentation map as prior information to guide the sparse self-attention in filtering and focusing on the correct regions.

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

↑