From ac517679e7b8b8449347b409710cef3137d67bc1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Wed, 17 Aug 2022 12:36:37 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index d27784de..2fcf7906 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -343,7 +343,7 @@ class ESPCN:
activation = tf.nn.leaky_relu
momentum = 0.99
- num_filters = 32
+ num_filters = 128
input_2d = self.inputs[0]
print('input: ', input_2d.shape)
@@ -366,6 +366,8 @@ class ESPCN:
conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4')
+ conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5')
+
conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding)(conv_b)
conv = conv + conv_b
--
GitLab