/DnCNN-PT-Lightning

PyTorch Lightning implementation of DnCNN (Denoising Convolution Neural Network)

Primary LanguageJupyter Notebook

DnCNN-PT-Lightning

  • Simple PyTorch Lightning implementation of DnCNN (Denoising Convolution Neural Network).

  • The notebook has been adapted from this awesome repository by lychengr3x.

  • Relatively large codebase shrunk down to a single notebook.

Implements a residual learning approach to denoise an image based on the BSDS-100 Dataset. Can be directly used in Colab.