test...

modules/deeplearning/espcn.py

@@ -195,14 +195,16 @@ class ESPCN:
         label = np.concatenate(label_s)
 
         # label = label[:, label_idx, :, :]
-        label = label[:, label_idx, 3:67, 3:67]
+        #label = label[:, label_idx, 3:67, 3:67]
+        label = label[:, label_idx, 0:32, 0:32]
         label = np.expand_dims(label, axis=3)
-        label = tf.image.resize(label, (32, 32), method='nearest').numpy()
+        #label = tf.image.resize(label, (32, 32), method='nearest').numpy()
 
         # data = data[:, data_idx, :, :]
-        data = data[:, data_idx, 3:67, 3:67]
+        #data = data[:, data_idx, 3:67, 3:67]
+        data = data[:, data_idx, 0:32, 0:32]
         data = np.expand_dims(data, axis=3)
-        data = tf.image.resize(data, (32, 32), method='nearest').numpy()
+        #data = tf.image.resize(data, (32, 32), method='nearest').numpy()
         # data = tf.image.resize(data, (36, 36)).numpy()
 
         data = data.astype(np.float32)