From f1df60e6b67d2767b1b23c1323db9811e732cb13 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 18 Aug 2022 09:49:47 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 2fcf7906..024f1aae 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -67,12 +67,13 @@ x_134_2 = x_134[2:133:2]
y_134_2 = y_134[2:133:2]
-def build_residual_conv2d_block(conv, num_filters, block_name, activation=tf.nn.leaky_relu, padding='SAME'):
+def build_residual_conv2d_block(conv, num_filters, block_name, activation=tf.nn.leaky_relu, padding='SAME', scale=None):
with tf.name_scope(block_name):
skip = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
skip = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=None)(skip)
-
+ if scale is not None:
+ skip = tf.keras.layers.Lambda(lambda x: x * scale)(skip)
conv = conv + skip
print(conv.shape)
--
GitLab