PD-Denoising-pytorch
PD-Denoising-pytorch copied to clipboard
bugs while using k>0 and pre-set stride mode(ps=2)
Hi,
Thanks for your great work on the real-world denoising! It gives me lots of intuitions.
It's a simple bug while using non-zero k and pre-set stride mode(ps=2). The oversmoothed image does not be reverse pixel-shuffled.
I think the problem on the N173 in test.py : max_out_numpy = visual_va2np(max_Out, opt.color, opt.ps, pss, 1, opt.rescale, w, h, c)
I changed it to: max_out_numpy = visual_va2np(max_Out, opt.color, opt.ps==1 or opt.ps==2, pss, 1, opt.rescale, w, h, c)
Is this right modification for your intend?
Thank you
Thanks!