Couldn't run main_train.py (keras) due to an error caused by datagenerator.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

Closed wlf114 closed 5 years ago

wlf114 commented 5 years ago

data = data.reshape((data.shape[0]*data.shape[1],data.shape[2],data.shape[3],1))

IndexError: tuple index out of range

wlf114 commented 5 years ago

The data file is missing the Train400 dataset.