From 1d4583fcf5090bb8ed5260c49d32f3718afa6d9f Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 8 Aug 2022 14:10:35 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 6 +-----
1 file changed, 1 insertion(+), 5 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index d189b523..4c3ba51d 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -433,11 +433,7 @@ class ESPCN:
# conv = tf.keras.layers.Conv2D(num_filters // 2, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
# print(conv.shape)
- conv = tf.keras.layers.Conv2D(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=2, padding=padding, activation=activation)(conv)
- conv = tf.keras.layers.Conv2DTranspose(1, kernel_size=3, strides=2, padding=padding, activation=activation)(conv)
+ 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)
--
GitLab