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

Adapter-BERT: Parameter-Efficient Transfer Learning for NLP.

A clean and scalable template to kickstart your deep learning project 🚀 ⚡ 🔥

Athena is the only tool that you will ever need to optimize your portfolio.

An Open-Source Package for Information Retrieval.

Rotation Robust Descriptors

A deep learning CNN model to identify and classify and check if a person is wearing a mask or not.

On-device speech-to-index engine powered by deep learning.

Demo code for paper "Learning optical flow from still images", CVPR 2021.

Repository for "Exploring Sparsity in Image Super-Resolution for Efficient Inference", CVPR 2021

YOLOv5 Series Multi-backbone, Pruning and quantization Compression Tool Box.

The original implementation of TNDM used in the NeurIPS 2021 paper (no longer being updated)

Create time-series datacubes for supervised machine learning with ICEYE SAR images.

Pixel-Perfect Structure-from-Motion with Featuremetric Refinement (ICCV 2021, Oral)

Lightweight Face Image Quality Assessment

Notes, programming assignments and quizzes from all courses within the Coursera Deep Learning specialization offered by deeplearning.ai

Deep and online learning with spiking neural networks in Python

NeurIPS workshop paper 'Counter-Strike Deathmatch with Large-Scale Behavioural Cloning'

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression

torchsummaryDynamic: support real FLOPs calculation of dynamic network or user-custom PyTorch ops

CRLT: A Unified Contrastive Learning Toolkit for Unsupervised Text Representation Learning