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

Learning Versatile Neural Architectures by Propagating Network Codes

NVIDIA Deep Learning Examples for Tensor Cores

Supervised Contrastive Learning for Product Matching

Steer OpenAI's Jukebox with Music Taggers

Traductor de lengua de señas al español basado en Python con Opencv y MedaiPipe

Official PyTorch implementation of Spatial Dependency Networks.

Head and Neck Tumour Segmentation and Prediction of Patient Survival Project

tensorflow code for inverse face rendering

Custom IMDB Dataset is extracted between 2020-2021 and custom distilBERT model is trained for movie success probability prediction

We present a regularized self-labeling approach to improve the generalization and robustness properties of fine-tuning.

A Keras implementation of YOLOv3 (Tensorflow backend)

Subpopulation detection in high-dimensional single-cell data

Codes for CVPR2021 paper "PWCLO-Net: Deep LiDAR Odometry in 3D Point Clouds Using Hierarchical Embedding Mask Optimization"

Official code of Team Yao at Multi-Modal-Fact-Verification-2022

PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)

Ranking Models in Unlabeled New Environments （iccv21）

CS550 Machine Learning course project on CNN Detection.

Towards Interpretable Deep Metric Learning with Structural Matching

Detecting drunk people through thermal images using Deep Learning (CNN)

Toolbox of models, callbacks, and datasets for AI/ML researchers.