Selective Wavelet Attention Learning for Single Image Deraining

Last update: Jun 17, 2022

Overview

SWAL

Code for Paper "Selective Wavelet Attention Learning for Single Image Deraining"

Prerequisites

Python 3
PyTorch

Models

We provide the models trained on DDN, DID, Rain100H, Rain100L, and AGAN datasets in the following links:

Download them into the model folder before testing.

Dataset

Download the rain datasets.
Arrange the images and generate a list file, just like the rain12 set in the data folder.

You can also modify the data_loader code in your manner.

Run

Train SWAL on a single GPU:

 CUDA_VISIBLE_DEVICES=0 python main.py --ngf=16 --ndf=64  --output_height=320  --trainroot=YOURPATH --trainfiles='YOUR_FILELIST'  --save_iter=1 --batchSize=8 --nrow=8 --lr_d=1e-4 --lr_g=1e-4  --cuda  --nEpochs=500

Train SWAL on multiple GPUs:

 CUDA_VISIBLE_DEVICES=0,1,2,3 python main.py --ngf=16 --ndf=64  --output_height=320  --trainroot=YOURPATH --trainfiles='YOUR_FILELIST'  --save_iter=1 --batchSize=32 --nrow=8 --lr_d=1e-4 --lr_g=1e-4  --cuda  --nEpochs=500

Test SWAL:

 CUDA_VISIBLE_DEVICES=0 python test.py --ngf=16  --outf='test' --testroot='data/rain12_test' --testfiles='data/rain12_test.list' --pretrained='model/rain100l_best.pth'  --cuda

Adjust the parameters according to your own settings.

Citation

If you use our codes, please cite the following paper:

 @article{huang2021selective,
   title={Selective Wavelet Attention Learning for Single Image Deraining},
   author={Huang, Huaibo and Yu, Aijing and Chai, Zhenhua and He, Ran and Tan, Tieniu},
   journal={International Journal of Computer Vision},
   volume={129},
   number={4},
   pages={1282--1300},
   year={2021},
  }

The released codes are only allowed for non-commercial use.

Selective Wavelet Attention Learning for Single Image Deraining

Related tags

Overview

SWAL

Prerequisites

Models

Dataset

Run

Citation

Owner

Bobo

Charsiu: A transformer-based phonetic aligner

[ICLR2021oral] Rethinking Architecture Selection in Differentiable NAS

Implementation of our NeurIPS 2021 paper "A Bi-Level Framework for Learning to Solve Combinatorial Optimization on Graphs".

[ICME 2021 Oral] CORE-Text: Improving Scene Text Detection with Contrastive Relational Reasoning

Repository for the paper "Online Domain Adaptation for Occupancy Mapping", RSS 2020

Pathdreamer: A World Model for Indoor Navigation

Expressive Power of Invariant and Equivaraint Graph Neural Networks (ICLR 2021)

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

Open-source implementation of Google Vizier for hyper parameters tuning

Project Tugas Besar pertama Pengenalan Komputasi Institut Teknologi Bandung

Music Generation using Neural Networks Streamlit App

deep learning model that learns to code with drawing in the Processing language

Directed Greybox Fuzzing with AFL

Outlier Exposure with Confidence Control for Out-of-Distribution Detection

StarGAN-ZSVC: Unofficial PyTorch Implementation

Test-Time Personalization with a Transformer for Human Pose Estimation, NeurIPS 2021

scAR (single-cell Ambient Remover) is a package for data denoising in single-cell omics.

A Strong Baseline for Image Semantic Segmentation

The Generic Manipulation Driver Package - Implements a ROS Interface over the robotics toolbox for Python

RMNA: A Neighbor Aggregation-Based Knowledge Graph Representation Learning Model Using Rule Mining