Code Monkey home page Code Monkey logo

wdenet's Introduction

WDENet: Wavelet-based Detail Enhanced Image Denoising Network

paper slides python pytorch

News

  • Jun 23, 2021: Outstanding Student Paper at Summer Conference of The Korean Institute of Broadcast and Media Engineers
  • Nov 30, 2021: Paper accepted at Journal of Broadcast Engineering

Abstract: Although the performance of cameras is gradualy improving now, there are noise in the acquired digital images from the camera, which acts as an obstacle to obtaining high-resolution images. Traditonaly, a filtering method has ben used for denoising, and a convolutional neural network (CNN), one of the dep learning techniques, has ben showing beter performance than traditonal methods in the field of image denoising, but the details in images could be lost during the learning proces. In this paper, we present a CNN for image denoising, which improves image details by learning the details of the image based on wavelet transform. The proposed network uses two subnetworks for detail enhancement and noise extraction. The experiment was conducted through Gausian noise and real-world noise, we confirmed that our proposed method was able to solve the detail los problem more efectively than conventional algorithms, and we verifed that both objective quality evaluation and subjective quality comparison showed excelent results.


The paper can be found here.

Network Architecture

proposed_model

Training and Evaluation

The used training and testing datasets and visual results can be downloaded as follows:

Task Training Datasets Testing Datasets WDENet's Pre-trained Models WDENet's Visual Results
Gaussian Grayscale Image Denoising DIV2K Set12, BSD68 Download Here
Real Image Denoising SIDD-Medium Dataset SIDD Validation Dataset Download Here

Instructions

Task Training Instructions Testing Instructions
Gaussian Grayscale Image Denoising Link Link
Real Image Denoising Link Link

Results

Experiments are performed for different image restoration tasks including, gaussian grayscale denoising and real image denoising.

PSNR and SSIM scores

Gaussian Grayscale Image Denoising
grayscale_image_denoising_psnr_ssim.jpg
Real Image Denoisng
real_image_denoising_psnr_ssim.jpg

Visual results

Gaussian Grayscale Image Denoising
  • Gaussian grayscale image denoising results of Set12 with noise level 15.
grayscale_image_denoising_result-1.jpg
  • Gaussian grayscale image denoising results of Set12 with noise level 25.
grayscale_image_denoising_result-2.jpg
  • Gaussian grayscale image denoising results of BSD68 with noise level 50.
grayscale_image_denoising_result-3.jpg
Real Image Denoisng
  • Real image denoising results of SIDD validation dataset.
real_image_denoising_result-1.jpg real_image_denoising_result-2.jpg

Run Time

  • Run time (in seconds) of different methods on grayscale images of size 256 × 256 , 512 × 512 and 1024 × 1024 with noise level 25.

run_time.jpg

Pre-trained models of WDENet can be downloaded here.

Contact

Should you have any question, please contact [email protected].

wdenet's People

Contributors

ggoo-un avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.