Skip to content

ggoo-un/WDENet

Repository files navigation

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].

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages