From 42377f83bd27aacdadfc32ac3a09c3898039dcad Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 18 Aug 2022 10:29:46 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 14 ++++++++------
1 file changed, 8 insertions(+), 6 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 024f1aae..297a3a3a 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -344,7 +344,7 @@ class ESPCN:
activation = tf.nn.leaky_relu
momentum = 0.99
- num_filters = 128
+ num_filters = 64
input_2d = self.inputs[0]
print('input: ', input_2d.shape)
@@ -359,15 +359,17 @@ class ESPCN:
if NOISE_TRAINING:
conv = conv_b = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv)
- conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_1')
+ scale = None
- conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_2')
+ conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_1', scale=scale)
- conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_3')
+ conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_2', scale=scale)
- 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_3', scale=scale)
- conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5')
+ conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_4', scale=scale)
+
+ conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_5', scale=scale)
conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding)(conv_b)
--
GitLab