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

Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"

Face Mask Detection is a project to determine whether someone is wearing mask or not, using deep neural network.

Gradient Inversion with Generative Image Prior

PyTorch GPU implementation of the ES-RNN model for time series forecasting

Experiments with the Robust Binary Interval Search (RBIS) algorithm, a Query-Based prediction algorithm for the Online Search problem.

IJON is an annotation mechanism that analysts can use to guide fuzzers such as AFL.

Code for Private Recommender Systems: How Can Users Build Their Own Fair Recommender Systems without Log Data? (SDM 2022)

StarGAN - Official PyTorch Implementation (CVPR 2018)

Very large and sparse networks appear often in the wild and present unique algorithmic opportunities and challenges for the practitioner

Hyperparameter tuning for humans

Official implementation of "Can You Spot the Chameleon? Adversarially Camouflaging Images from Co-Salient Object Detection" in CVPR 2022.

Source code for PairNorm (ICLR 2020)

EMNLP 2021: Single-dataset Experts for Multi-dataset Question-Answering

Elastic weight consolidation technique for incremental learning.

Deep Learning as a Cloud API Service.

ANEA: Distant Supervision for Low-Resource Named Entity Recognition

Resources related to EMNLP 2021 paper "FAME: Feature-Based Adversarial Meta-Embeddings for Robust Input Representations"

PyTorch implementation of the Deep SLDA method from our CVPRW-2020 paper "Lifelong Machine Learning with Deep Streaming Linear Discriminant Analysis"

MemStream: Memory-Based Anomaly Detection in Multi-Aspect Streams with Concept Drift

Efficient and intelligent interactive segmentation annotation software