From d09abe7ebe9266222ead827b751052f4ba050c84 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 2 Aug 2022 14:31:31 -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 06abdbfd..2890ce25 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -212,11 +212,11 @@ class ESPCN:
self.n_chans = 1
# self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
- self.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
+ self.X_img = tf.keras.Input(shape=(36, 36, self.n_chans))
self.inputs.append(self.X_img)
# self.inputs.append(tf.keras.Input(shape=(None, None, self.n_chans)))
- self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
+ self.inputs.append(tf.keras.Input(shape=(36, 36, self.n_chans)))
self.DISK_CACHE = False
@@ -411,7 +411,7 @@ class ESPCN:
input_2d = self.inputs[0]
print('input: ', input_2d.shape)
conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding='VALID', activation=None)(input_2d)
- conv = conv[:, 4:20, 4:20, :]
+ # conv = conv[:, 4:20, 4:20, :]
print('Contracting Branch')
print('input: ', conv.shape)
skip = conv
--
GitLab