Skip to content

Pythonista7/DnCNN-tf2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Deep Denoise

This is a tensorflow-2 implementation of the paper Beyond a Gaussian Denoiser: Residual Learning ofDeep CNN for Image Denoising.

Dataset used :

BSDS200

To download and unpack run:

wget https://www2.eecs.berkeley.edu/Research/Projects/CS/vision/bsds/BSDS300-images.tgz

tar -xvzf BSDS300-images.tgz

Requriments :

  • tensorflow==2.3.1
  • matplotlib
  • numpy

Run Model

Run train_denoise.py

Results

After training by setting depth=5 for 25 epoch we get the below results. The original paper suggests a deeper architecture feel free to tweak the hyper params in the train_denoise.py file.

alt

alt