SDNet: mutil-branch for single image deraining using swin
- URL: http://arxiv.org/abs/2105.15077v1
- Date: Mon, 31 May 2021 16:06:02 GMT
- Title: SDNet: mutil-branch for single image deraining using swin
- Authors: Fuxiang Tan, YuTing Kong, Yingying Fan, Feng Liu, Daxin Zhou, Hao
zhang, Long Chen, Liang Gao and Yurong Qian
- Abstract summary: We introduce Swin-transformer into the field of image deraining for the first time.
Specifically, we improve the basic module of Swin-transformer and design a three-branch model to implement single-image rain removal.
Our proposed method has performance and inference speed advantages over the current mainstream single-image rain streaks removal models.
- Score: 14.574622548559269
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Rain streaks degrade the image quality and seriously affect the performance
of subsequent computer vision tasks, such as autonomous driving, social
security, etc. Therefore, removing rain streaks from a given rainy images is of
great significance. Convolutional neural networks(CNN) have been widely used in
image deraining tasks, however, the local computational characteristics of
convolutional operations limit the development of image deraining tasks.
Recently, the popular transformer has global computational features that can
further facilitate the development of image deraining tasks. In this paper, we
introduce Swin-transformer into the field of image deraining for the first time
to study the performance and potential of Swin-transformer in the field of
image deraining. Specifically, we improve the basic module of Swin-transformer
and design a three-branch model to implement single-image rain removal. The
former implements the basic rain pattern feature extraction, while the latter
fuses different features to further extract and process the image features. In
addition, we employ a jump connection to fuse deep features and shallow
features. In terms of experiments, the existing public dataset suffers from
image duplication and relatively homogeneous background. So we propose a new
dataset Rain3000 to validate our model. Therefore, we propose a new dataset
Rain3000 for validating our model. Experimental results on the publicly
available datasets Rain100L, Rain100H and our dataset Rain3000 show that our
proposed method has performance and inference speed advantages over the current
mainstream single-image rain streaks removal models.The source code will be
available at https://github.com/H-tfx/SDNet.
Related papers
- RainyScape: Unsupervised Rainy Scene Reconstruction using Decoupled Neural Rendering [50.14860376758962]
We propose RainyScape, an unsupervised framework for reconstructing clean scenes from a collection of multi-view rainy images.
Based on the spectral bias property of neural networks, we first optimize the neural rendering pipeline to obtain a low-frequency scene representation.
We jointly optimize the two modules, driven by the proposed adaptive direction-sensitive gradient-based reconstruction loss.
arXiv Detail & Related papers (2024-04-17T14:07:22Z) - From Sky to the Ground: A Large-scale Benchmark and Simple Baseline
Towards Real Rain Removal [28.029107707930063]
We construct a Large-scale High-quality Paired real rain benchmark (LHP-Rain), including 3000 video sequences with 1 million high-resolution (1920*1080) frame pairs.
The advantages of the proposed dataset are three-fold: rain with higher-diversity and larger-scale, image with higher-resolution and higher-quality ground-truth.
arXiv Detail & Related papers (2023-08-07T18:39:14Z) - Single Image Deraining via Feature-based Deep Convolutional Neural
Network [13.39233717329633]
A single image deraining algorithm based on the combination of data-driven and model-based approaches is proposed.
Experiments show that the proposed algorithm significantly outperforms state-of-the-art methods in terms of both qualitative and quantitative measures.
arXiv Detail & Related papers (2023-05-03T13:12:51Z) - Toward Real-world Single Image Deraining: A New Benchmark and Beyond [79.5893880599847]
Single image deraining (SID) in real scenarios attracts increasing attention in recent years.
Previous real datasets suffer from low-resolution images, homogeneous rain streaks, limited background variation, and even misalignment of image pairs.
We establish a new high-quality dataset named RealRain-1k, consisting of $1,120$ high-resolution paired clean and rainy images with low- and high-density rain streaks, respectively.
arXiv Detail & Related papers (2022-06-11T12:26:59Z) - Semi-DRDNet Semi-supervised Detail-recovery Image Deraining Network via
Unpaired Contrastive Learning [59.22620253308322]
We propose a semi-supervised detail-recovery image deraining network (termed as Semi-DRDNet)
As a semi-supervised learning paradigm, Semi-DRDNet operates smoothly on both synthetic and real-world rainy data in terms of deraining robustness and detail accuracy.
arXiv Detail & Related papers (2022-04-06T12:35:27Z) - RCDNet: An Interpretable Rain Convolutional Dictionary Network for
Single Image Deraining [49.99207211126791]
We specifically build a novel deep architecture, called rain convolutional dictionary network (RCDNet)
RCDNet embeds the intrinsic priors of rain streaks and has clear interpretability.
By end-to-end training such an interpretable network, all involved rain kernels and proximal operators can be automatically extracted.
arXiv Detail & Related papers (2021-07-14T16:08:11Z) - Beyond Monocular Deraining: Parallel Stereo Deraining Network Via
Semantic Prior [103.49307603952144]
Most existing de-rain algorithms use only one single input image and aim to recover a clean image.
We present a Paired Rain Removal Network (PRRNet), which exploits both stereo images and semantic information.
Experiments on both monocular and the newly proposed stereo rainy datasets demonstrate that the proposed method achieves the state-of-the-art performance.
arXiv Detail & Related papers (2021-05-09T04:15:10Z) - From Rain Generation to Rain Removal [67.71728610434698]
We build a full Bayesian generative model for rainy image where the rain layer is parameterized as a generator.
We employ the variational inference framework to approximate the expected statistical distribution of rainy image.
Comprehensive experiments substantiate that the proposed model can faithfully extract the complex rain distribution.
arXiv Detail & Related papers (2020-08-08T18:56:51Z) - Multi-Task Learning Enhanced Single Image De-Raining [9.207797392774465]
Rain removal in images is an important task in computer vision filed and attracting attentions of more and more people.
In this paper, we address a non-trivial issue of removing visual effect of rain streak from a single image.
Our method combines various semantic constraint task in a proposed multi-task regression model for rain removal.
arXiv Detail & Related papers (2020-03-21T16:19:56Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.