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

Code for the ICCV2021 paper "Personalized Image Semantic Segmentation"

Deeplab-resnet-101 in Pytorch with Jaccard loss

YOLOv7 - Framework Beyond Detection

Face2webtoon - Despite its importance, there are few previous works applying I2I translation to webtoon.

Final Project for the CS238: Decision Making Under Uncertainty course at Stanford University in Autumn '21.

we propose a novel deep network, named feature aggregation and refinement network (FARNet), for the automatic detection of anatomical landmarks.

alfred-py: A deep learning utility library for human

Official implementations of PSENet, PAN and PAN++.

CFNet: Cascade and Fused Cost Volume for Robust Stereo Matching（CVPR2021）

FADNet++: Real-Time and Accurate Disparity Estimation with Configurable Networks

A curated list of references for MLOps

Anchor-free Oriented Proposal Generator for Object Detection

Graph Convolutional Networks for Temporal Action Localization (ICCV2019)

Unofficial Implementation of MLP-Mixer in TensorFlow

Semantically Contrastive Learning for Low-light Image Enhancement

GPU implementation of $k$-Nearest Neighbors and Shared-Nearest Neighbors

GDR-Net: Geometry-Guided Direct Regression Network for Monocular 6D Object Pose Estimation. (CVPR 2021)

Graph Convolutional Networks in PyTorch

GRaNDPapA: Generator of Rad Names from Decent Paper Acronyms

Implementation of RegretNet with Pytorch

Selective Wavelet Attention Learning for Single Image Deraining

Related tags

Overview

SWAL

Prerequisites

Models

Dataset

Run

Citation

Owner

Bobo

Code for the ICCV2021 paper "Personalized Image Semantic Segmentation"

Deeplab-resnet-101 in Pytorch with Jaccard loss

YOLOv7 - Framework Beyond Detection

Face2webtoon - Despite its importance, there are few previous works applying I2I translation to webtoon.

Final Project for the CS238: Decision Making Under Uncertainty course at Stanford University in Autumn '21.

we propose a novel deep network, named feature aggregation and refinement network (FARNet), for the automatic detection of anatomical landmarks.

alfred-py: A deep learning utility library for **human**

Official implementations of PSENet, PAN and PAN++.

CFNet: Cascade and Fused Cost Volume for Robust Stereo Matching（CVPR2021）

FADNet++: Real-Time and Accurate Disparity Estimation with Configurable Networks

A curated list of references for MLOps

Anchor-free Oriented Proposal Generator for Object Detection

Graph Convolutional Networks for Temporal Action Localization (ICCV2019)

Unofficial Implementation of MLP-Mixer in TensorFlow

Semantically Contrastive Learning for Low-light Image Enhancement

GPU implementation of $k$-Nearest Neighbors and Shared-Nearest Neighbors

GDR-Net: Geometry-Guided Direct Regression Network for Monocular 6D Object Pose Estimation. (CVPR 2021)

Graph Convolutional Networks in PyTorch

GRaNDPapA: Generator of Rad Names from Decent Paper Acronyms

Implementation of RegretNet with Pytorch

alfred-py: A deep learning utility library for human