diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 1ab09707bab8af737f859821abb0326a4ab54feb..54e15bf5214f9ba9fa8aa73d18d1ee62bfee8458 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -733,7 +733,7 @@ class ESPCN: def prepare(param_idx=1, filename='/Users/tomrink/data_valid_40.npy'): nda = np.load(filename) #nda = nda[:, param_idx, :, :] - nda_lr = nda[:, param_idx, x_134_2, y_134_2] + nda_lr = nda[:, param_idx, 2:133:2, 2:133:2] # nda_lr = resample(x_70, y_70, nda, x_70_2, y_70_2) nda_lr = np.expand_dims(nda_lr, axis=3) return nda_lr