From 6f5ffcea17f455c5c0f138092b00e47c692cdb4b Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sat, 20 Aug 2022 16:34:55 -0500
Subject: [PATCH] snapshot..
---
modules/deeplearning/espcn.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 29ef22b6..5e78a128 100644
--- a/modules/deeplearning/espcn.py
+++ b/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()
--
GitLab