snapshot...

modules/deeplearning/espcn.py

@@ -196,7 +196,7 @@ class ESPCN:
         # label = label[:, label_idx, :, :]
         label = label[:, label_idx, 3:67, 3:67]
         label = np.expand_dims(label, axis=3)
-        label = tf.image.resize(label, (32, 32))
+        label = tf.image.resize(label, (32, 32)).numpy()
 
         # data = data[:, data_idx, :, :]
         data = data[:, data_idx, 3:67, 3:67]