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

The pyrelational package offers a flexible workflow to enable active learning with as little change to the models and datasets as possible

Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training of neural networks"

Repo for the paper "DiLBERT: Cheap Embeddings for Disease Related Medical NLP"

Single cell current best practices tutorial case study for the paper:Luecken and Theis, "Current best practices in single-cell RNA-seq analysis: a tutorial"

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow

hipCaffe: the HIP port of Caffe

Semi-supervised learning for object detection

Wafer Fault Detection using MlOps Integration

Zero-Cost Proxies for Lightweight NAS

Notebooks for my "Deep Learning with TensorFlow 2 and Keras" course

Neural Tangent Generalization Attacks (NTGA)

UnivNet: A Neural Vocoder with Multi-Resolution Spectrogram Discriminators for High-Fidelity Waveform Generation

Code for "Retrieving Black-box Optimal Images from External Databases" (WSDM 2022)

Official implementation for paper Knowledge Bridging for Empathetic Dialogue Generation (AAAI 2021).

A vision library for performing sliced inference on large images/small objects

根据midi文件演奏“风物之诗琴”的脚本 "Windsong Lyre" auto play

Pytorch implementation of the paper Progressive Growing of Points with Tree-structured Generators (BMVC 2021)

Kaggle | 9th place (part of) solution for the Bristol-Myers Squibb – Molecular Translation challenge

Gauge equivariant mesh cnn

In generative deep geometry learning, we often get many obj files remain to be rendered