From e37ca30f31e578b7714940e81ffb36372db68b3c Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 8 Aug 2022 14:41:31 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 4c3ba51d..990036ec 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -436,11 +436,11 @@ class ESPCN:
conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=4, strides=2, padding=padding, activation=activation)(conv)
print(conv.shape)
- # conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
- # print(conv.shape)
- #
- # conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
- # print(conv.shape)
+ conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=1, strides=1, padding=padding, activation=activation)(conv)
+ print(conv.shape)
+
+ conv = tf.keras.layers.Conv2DTranspose(num_filters // 8, kernel_size=1, strides=1, padding=padding, activation=activation)(conv)
+ print(conv.shape)
#self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability', activation=tf.nn.sigmoid)(conv)
self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability')(conv)
--
GitLab