GuoShi28/CBDNet

Noise estimation map

jarvislkm opened this issue · 2 comments

Hi, is it possible to show one or two output image of the noise estimation network? For both synthesis image and real world image? Thanks.

@jarvislkm Hi, Sorry for getting back to you so late. You can obtain noise estimation by adding the following code in the released test code:

noise_est = net.getVarIndex('predictionSigma');
net.vars(net.getVarIndex('predictionSigma')).precious = 1;

run CNN denoiser

est_sigma = squeeze(gather(net.vars(noise_est).value));

If you have other suggestions or questions, please feel free to contact me.

Thanks, it works.