English

SUNet: Swin Transformer UNet for Image Denoising

Image and Video Processing 2023-01-18 v1 Computer Vision and Pattern Recognition Machine Learning

Abstract

Image restoration is a challenging ill-posed problem which also has been a long-standing issue. In the past few years, the convolution neural networks (CNNs) almost dominated the computer vision and had achieved considerable success in different levels of vision tasks including image restoration. However, recently the Swin Transformer-based model also shows impressive performance, even surpasses the CNN-based methods to become the state-of-the-art on high-level vision tasks. In this paper, we proposed a restoration model called SUNet which uses the Swin Transformer layer as our basic block and then is applied to UNet architecture for image denoising. The source code and pre-trained models are available at https://github.com/FanChiMao/SUNet.

Keywords

Cite

@article{arxiv.2202.14009,
  title  = {SUNet: Swin Transformer UNet for Image Denoising},
  author = {Chi-Mao Fan and Tsung-Jung Liu and Kuan-Hsien Liu},
  journal= {arXiv preprint arXiv:2202.14009},
  year   = {2023}
}