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

基于Swin-Conv-UNet与数据合成的实用盲图像去噪

Practical Blind Image Denoising via Swin-Conv-UNet and Data Synthesis

ETH Zürich(苏黎世联邦理工学院);University of Würzburg(维尔茨堡大学);KU Leuven(鲁汶大学)

Kai Zhang, Yawei Li, Jingyun Liang, Jiezhang Cao, Yulun Zhang, Hao Tang, Deng-Ping Fan, Radu Timofte, Luc Van Gool

AI总结:

本文提出Swin-Conv-UNet架构,结合卷积局部建模与Swin Transformer非局部建模能力,并设计包含多种噪声与双重退化的数据合成模型,显著提升真实图像盲去噪的实用性与性能。

摘要:

尽管近年来利用深度神经网络解决图像去噪问题出现了急剧增长,但现有方法大多依赖简单的噪声假设,例如加性高斯白噪声(AWGN)、JPEG压缩噪声和相机传感器噪声,而面向真实图像的通用盲去噪方法仍未得到解决。本文尝试从网络架构设计和训练数据合成两个角度来解决这一问题。具体而言,在网络架构设计方面,我们提出了一种swin-conv模块,以融合残差卷积层的局部建模能力和swin transformer模块的非局部建模能力,并将其作为主要构建模块嵌入广泛使用的图像到图像转换UNet架构中。在训练数据合成方面,我们设计了一种实用的噪声退化模型,该模型考虑了不同类型的噪声(包括高斯噪声、泊松噪声、散斑噪声、JPEG压缩噪声和处理后的相机传感器噪声)以及尺寸调整,并引入了随机混洗策略和双重退化策略。在AWGN去除和真实图像去噪上的大量实验表明,新的网络架构设计达到了最先进的性能,并且新的退化模型能够显著提升实用性。我们相信我们的工作可以为当前的去噪研究提供有益的启示。

英文摘要:

While recent years have witnessed a dramatic upsurge of exploiting deep neural networks toward solving image denoising, existing methods mostly rely on simple noise assumptions, such as additive white Gaussian noise (AWGN), JPEG compression noise and camera sensor noise, and a general-purpose blind denoising method for real images remains unsolved. In this paper, we attempt to solve this problem from the perspective of network architecture design and training data synthesis. Specifically, for the network architecture design, we propose a swin-conv block to incorporate the local modeling ability of residual convolutional layer and non-local modeling ability of swin transformer block, and then plug it as the main building block into the widely-used image-to-image translation UNet architecture. For the training data synthesis, we design a practical noise degradation model which takes into consideration different kinds of noise (including Gaussian, Poisson, speckle, JPEG compression, and processed camera sensor noises) and resizing, and also involves a random shuffle strategy and a double degradation strategy. Extensive experiments on AGWN removal and real image denoising demonstrate that the new network architecture design achieves state-of-the-art performance and the new degradation model can help to significantly improve the practicability. We believe our work can provide useful insights into current denoising research.

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

↑