it seems that there are some preblems in the pytorch implement of this code!

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

it seems that there are some preblems in the pytorch implement of this code! #75

Open cscss opened 4 years ago

cscss commented 4 years ago

Hi, Thanks for release the code. I have read your code. It's brilliant! but I found a small problem in your pytorch code. when get patches. you data augment have some problems about cv2.imresize. you can check you codes in ./DnCNN-master/TrainingCodes/dncnn_pytorch/data_generator.py line 94.

oneTaken commented 4 years ago

What's the problem?

cscss commented 4 years ago

When you use opencv to bicubic, the size of image should be (w,h), not (h, w). It is just a small problem, but it could be error when run this code.

oneTaken commented 4 years ago

Yeah, you're right. It's really a small code bug.