snapshot..

modules/deeplearning/espcn.py

@@ -605,8 +605,8 @@ class ESPCN:
 
         self.reset_test_metrics()
 
-        for data0, data1, label in self.test_dataset:
-            ds = tf.data.Dataset.from_tensor_slices((data0, data1, label))
+        for data, label in self.test_dataset:
+            ds = tf.data.Dataset.from_tensor_slices((data, label))
             ds = ds.batch(BATCH_SIZE)
             for mini_batch_test in ds:
                 self.predict(mini_batch_test)
@@ -639,7 +639,7 @@ class ESPCN:
 
         self.reset_test_metrics()
 
-        pred = self.model([data])
+        pred = self.model([data], training=False)
         self.test_probs = pred
         pred = pred.numpy()