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

The official project of SimSwap (ACM MM 2020)

Voxel Set Transformer: A Set-to-Set Approach to 3D Object Detection from Point Clouds (CVPR 2022)

A simple log parser and summariser for IIS web server logs

Tianshou - An elegant PyTorch deep reinforcement learning library.

Short and long time series classification using convolutional neural networks

A Python training and inference implementation of Yolov5 helmet detection in Jetson Xavier nx and Jetson nano

Statistical and Algorithmic Investing Strategies for Everyone

Text Extraction Formulation + Feedback Loop for state-of-the-art WSD (EMNLP 2021)

This repository contains FEDOT - an open-source framework for automated modeling and machine learning (AutoML)

PyTorch implementation of "A Full-Band and Sub-Band Fusion Model for Real-Time Single-Channel Speech Enhancement."

MacroTools provides a library of tools for working with Julia code and expressions.

An implementation for the ICCV 2021 paper Deep Permutation Equivariant Structure from Motion.

Dynamic Capacity Networks using Tensorflow

GenshinMapAutoMarkTools - Tools To add/delete/refresh resources mark in Genshin Impact Map

Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track (SIGIR 2021 Full Paper).

An official source code for "Augmentation-Free Self-Supervised Learning on Graphs"

Unofficial implementation of the paper: PonderNet: Learning to Ponder in TensorFlow

An executor that performs image segmentation on fashion items

Kindle is an easy model build package for PyTorch.

DI-HPC is an acceleration operator component for general algorithm modules in reinforcement learning algorithms