diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 1b2ceeb281b68061cc2719693f46066251092a61..67b8edf6bf392fdb98e67dd3b60641e41d77a9f3 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -199,7 +199,7 @@ class ESPCN: label_s.append(nda) data = np.concatenate(label_s) - label = np.concatenate(label_s) + label = data.copy() data = data[:, data_idx, :, :] data = resample(x_70, y_70, data, x_70_2, y_70_2) @@ -732,7 +732,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, 3:67:2, 3:67:2] + nda_lr = nda[:, param_idx, 2:69:2, 2:69: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