From d05bc0bd8b25d985029fe594b20a14a4fbe0772c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 10 Aug 2022 14:54:35 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 40a79ee0..71ce8bfb 100644
--- a/modules/deeplearning/espcn.py
+++ b/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]
--
GitLab