Update data_generator.py

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

Open Gaochenyin opened 3 years ago

Gaochenyin commented 3 years ago

Switch the width and height in the cv2.resize when dealing with non-square images