├── EDA └── Images_EDA.ipynb ├── Final.ipynb ├── Models ├── (Baseline)_REDNet_256x256.ipynb ├── MWCNN_256x256.ipynb └── PRIDNet_256x256.ipynb ├── README.md ├── Results ├── PRIDNet_Prediction_1.png ├── PRIDNet_Prediction_2.png ├── PRIDNet_Prediction_3.png └── Result_scores.png └── Traditional Filter [NLM] └── Traditional_Filter_256x256.ipynb /README.md: -------------------------------------------------------------------------------- 1 | NOTE: In case you are not able to view the ipynb files on github, then paste its link on nbviewer, 2 | https://nbviewer.jupyter.org/ 3 | 4 | # Image-Denoising-using-Deep-Learning 5 | 6 | In this repo I have implemented three different deep learning architectures for image denoising,
7 | REDNet; https://arxiv.org/pdf/1606.08921.pdf
8 | Multi-level Wavelet CNN (MWCNN); https://arxiv.org/pdf/1805.07071.pdf
9 | PRIDNet; https://arxiv.org/pdf/1908.00273.pdf
10 | 11 | I found that PRIDNet was giving the best results as compared to the other two, giving me the best PSNR and SSIM scores along with best visualizations which were very close to the ground truth images.
12 | 13 | ![Screenshot](Results/Result_scores.png) 14 | 15 | ## PRIDNet Results 16 | 17 | ![Screenshot](Results/PRIDNet_Prediction_2.png) 18 | ![Screenshot](Results/PRIDNet_Prediction_1.png) 19 | ![Screenshot](Results/PRIDNet_Prediction_3.png) 20 | Notice the level of detail in the eye ball. 21 | -------------------------------------------------------------------------------- /Results/PRIDNet_Prediction_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chintan1995/Image-Denoising-using-Deep-Learning/7a98abd5ae8e131e64ba61fb4e0b546e71e5c237/Results/PRIDNet_Prediction_1.png -------------------------------------------------------------------------------- /Results/PRIDNet_Prediction_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chintan1995/Image-Denoising-using-Deep-Learning/7a98abd5ae8e131e64ba61fb4e0b546e71e5c237/Results/PRIDNet_Prediction_2.png -------------------------------------------------------------------------------- /Results/PRIDNet_Prediction_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chintan1995/Image-Denoising-using-Deep-Learning/7a98abd5ae8e131e64ba61fb4e0b546e71e5c237/Results/PRIDNet_Prediction_3.png -------------------------------------------------------------------------------- /Results/Result_scores.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chintan1995/Image-Denoising-using-Deep-Learning/7a98abd5ae8e131e64ba61fb4e0b546e71e5c237/Results/Result_scores.png --------------------------------------------------------------------------------