cszn / DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)
https://cszn.github.io/
1.42k stars 534 forks source link

Average PSNR vs epochs - denoised images with spots? #76

Open betty97 opened 4 years ago

betty97 commented 4 years ago

Hi, I am trying to reproduce the red graph provided in the paper which shows the Average PSNR vs epochs for the Set68: psnrvsepoch

I am using the Keras implementation provided here (it includes Residual Learning and Batch Normalization, and Adam optimization as well), and I am training 50 epochs.

However, instead of a stable graph like the one in the paper, I get a graph with pronounced drops, similar to the blue and black graphs in the paper. Furthermore, even in the last epochs, where the network should be converging, the Average PSNR often shows significant drops, which are related to the presence of odd spots in some of the denoised images: 01_dncnn test001_dncnn

I have experimented with different settings, in particular with different BN momentum, and the situation becomes better when it is set to a high value, as 0.9. Even so, the graph I get is never something similar to the red one in the paper. I am afraid these spots in the denoised images appear quite randomly, with different shapes and locations for different trainings.

Has anyone experienced the same or something similar? I would appreciate any help, thank you!!