From 90bf160f8165c06ceb3c8569dd814142ce60b35e Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 10 Aug 2022 14:36:53 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index b99843c6..40a79ee0 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -172,7 +172,7 @@ class ESPCN:
self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
# self.X_img = tf.keras.Input(shape=(36, 36, self.n_chans))
- # self.X_img = tf.keras.Input(shape=(32, 32, self.n_chans))
+ self.X_img = tf.keras.Input(shape=(32, 32, self.n_chans))
self.inputs.append(self.X_img)
@@ -196,6 +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))
# data = data[:, data_idx, :, :]
data = data[:, data_idx, 3:67, 3:67]
@@ -366,8 +367,8 @@ class ESPCN:
# conv = tf.keras.layers.Conv2D((factor ** 2), 3, padding='same')(conv)
print(conv.shape)
- conv = tf.nn.depth_to_space(conv, factor)
- print(conv.shape)
+ # conv = tf.nn.depth_to_space(conv, factor)
+ # print(conv.shape)
self.logits = tf.keras.layers.Conv2D(1, kernel_size=3, strides=1, padding=padding, name='regression')(conv)
--
GitLab