From b1365a2e98d4d80bbcfaee16b297f800cb3cf8f7 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 30 Aug 2022 22:18:49 -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 ccaa9842..d9df97b2 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -209,7 +209,8 @@ class ESPCN:
data = np.expand_dims(data, axis=3)
# label = label[:, label_idx, :, :]
- label = label[:, label_idx, 3:131:2, 3:131:2]
+ # label = label[:, label_idx, 3:131:2, 3:131:2]
+ label = label[:, label_idx, 3:131, 3:131]
# label = label[:, label_idx, 3:67, 3:67]
label = np.expand_dims(label, axis=3)
@@ -387,10 +388,10 @@ class ESPCN:
conv = conv + conv_b
print(conv.shape)
- # conv = tf.keras.layers.Conv2D(num_filters * (factor ** 2), 3, padding='same')(conv)
+ conv = tf.keras.layers.Conv2D(num_filters * (factor ** 2), 3, padding='same')(conv)
print(conv.shape)
- #conv = tf.nn.depth_to_space(conv, factor)
+ conv = tf.nn.depth_to_space(conv, factor)
#conv = tf.keras.layers.Conv2DTranspose(num_filters * (factor ** 2), 3, padding='same')(conv)
print(conv.shape)
--
GitLab