snapshot...

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