From 7c9bc2f1e8ff63d285d1955c33a0232e63dcc8ce Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 5 Apr 2022 15:12:48 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/unet.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 27fcc7bd..d0c23939 100644
--- a/modules/deeplearning/unet.py
+++ b/modules/deeplearning/unet.py
@@ -206,7 +206,8 @@ class UNET:
n_chans = len(self.train_params)
if TRIPLET:
n_chans *= 3
- self.X_img = tf.keras.Input(shape=(None, None, n_chans))
+ #self.X_img = tf.keras.Input(shape=(None, None, n_chans))
+ self.X_img = tf.keras.Input(shape=(16, 16, n_chans))
self.inputs.append(self.X_img)
self.inputs.append(tf.keras.Input(shape=(None, None, 5)))
@@ -604,6 +605,7 @@ class UNET:
conv = tf.keras.layers.LeakyReLU()(conv)
# Expanding (Decoding) branch -------------------------------------------------------------------------------
+ print('expanding branch')
num_filters /= 2
conv = tf.keras.layers.Conv2DTranspose(num_filters, kernel_size=3, strides=2, padding=padding)(conv)
--
GitLab