From 056669eb4d2ab9d2df08a0666a9c22e59058f81f Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 3 Aug 2022 12:24:43 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/espcn.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index f39d303c..821cf096 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -240,7 +240,9 @@ class ESPCN: nda = np.load(f) label_s.append(nda) - data = np.concatenate(data_s) + #data = np.concatenate(data_s) + data = np.concatenate(label_s) + data = tf.image.resize(data, (32, 32)) label = np.concatenate(label_s) label = label[:, label_idx, :, :] @@ -416,8 +418,8 @@ 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 = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding='VALID', activation=None)(input_2d) + conv = input_2d print('Contracting Branch') print('input: ', conv.shape) skip = conv -- GitLab