SDWNet: A Straight Dilated Network with Wavelet Transformation for Image Deblurring(offical)

1. Introduction

This repo is not only used for our paper(SDWNet) but also used for Deblur codebase. We implement a number of components that allow you to quickly implement your own model.

Paper The SDWNet has been accepted by iccvw2021, you can read the paper here.
Model

2. Folder Structure

  ---SDWNet
  |
  |- config
  |    |- model.yaml                            -> Model all traninig hyparameters with data log.
  |    |-Config.py                              -> Translate the config file to dict.
  |- data
  |    |- vanilar_dataset.py                    -> The dataset for build the LR & HR images.
  |    |- utils.py                              -> Utils for get patch and calculate the model metrics.
  |    |- augments.py                           -> Augment method for LR & HR images.
  |- model
  |    - NTIRE2021_Deblur
  |        - uniA_ELU
  |            |- layerlib_stage1               -> Model module.
  |            |- model_stage1_dual_branch_tail.py -> Main model.
  |- loss
  |   |- gendrator_loss.py                      -> Loss function define.
  |- optim
  |   |- optimizer.py                           -> Optimizer function define.
  |- train.py                                   -> Training.
  |- goprol_train.sh                             -> Training shell.
  |- inference_ddp.py                           -> Inference.
  |- inference_ddp.sh                           -> Inference shell.

3. Training

Crop the src Training LR and HR images to 480x480 by sliding window which step is 240, so we got 24 patchs form one 720x1280 images both LR and HR.
Training the model with the 416 x 416 size, use randomcrop, RGB shuffle, horizon flip, rotate and so on.
Normalize the images to Tensor with 255 but not 1. which without process the mean and std.

python -W ignore train.py \
--config_file $config_folder \
--dist-url 'tcp://127.0.0.1:8888' \
--dist-backend 'nccl' \
--multiprocessing-distributed=1 \
--world-size=1 \
--rank=0 \

4. Inference

Inference the src LR images and get the SR images

python -W ignore inference_ddp.py \
--config_file $config_folder \
--dist-url 'tcp://127.0.0.1:8989' \
--dist-backend 'nccl' \
--multiprocessing-distributed=1 \
--world-size=1 \
--rank=0 \

5. Calculate

Calculate the PSNR and SSIM

python utils/calc_psnr_ssim_official.py

If you find this repo useful for your research, please consider citing the papers

@InProceedings{
  Zou_2021_ICCV, 
  author = {Zou, Wenbin and Jiang, Mingchao and Zhang, Yunchen and Chen, Liang and Lu, Zhiyong and Wu, Yi}, 
  title = {SDWNet: A Straight Dilated Network With Wavelet Transformation for Image Deblurring}, 
  booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV) Workshops}, 
  month = {October}, 
  year = {2021}, 
  pages = {1895-1904} 
  }

An Straight Dilated Network with Wavelet for image Deblurring

Related tags

Overview

SDWNet: A Straight Dilated Network with Wavelet Transformation for Image Deblurring(offical)

1. Introduction

2. Folder Structure

3. Training

4. Inference

5. Calculate

Owner

FlyEgle

Super Resolution for images using deep learning.

HODEmu, is both an executable and a python library that is based on Ragagnin 2021 in prep.

Symbolic Parallel Adaptive Importance Sampling for Probabilistic Program Analysis in JAX

A modular application for performing anomaly detection in networks

Weakly-supervised semantic image segmentation with CNNs using point supervision

Azion the best solution of Edge Computing in the world.

Contextualized Perturbation for Textual Adversarial Attack, NAACL 2021

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

A general-purpose encoder-decoder framework for Tensorflow

Weakly supervised medical named entity classification

Stroke-predictions-ml-model - Machine learning model to predict individuals chances of having a stroke

PyTorch implementation of the REMIND method from our ECCV-2020 paper "REMIND Your Neural Network to Prevent Catastrophic Forgetting"

Spatial color quantization in Rust

Official repository of the AAAI'2022 paper "Contrast and Generation Make BART a Good Dialogue Emotion Recognizer"

Research on Tabular Deep Learning (Python package & papers)

a reimplementation of Optical Flow Estimation using a Spatial Pyramid Network in PyTorch

Image Classification - A research on image classification and auto insurance claim prediction, a systematic experiments on modeling techniques and approaches

Code for the head detector (HeadHunter) proposed in our CVPR 2021 paper Tracking Pedestrian Heads in Dense Crowd.

Road Crack Detection Using Deep Learning Methods

Best Practices on Recommendation Systems