From b71bdab59100f27ea58bfa95a583a0d7f06247a8 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 19 Aug 2022 11:39:30 -0500
Subject: [PATCH] snapshot..
---
modules/deeplearning/srcnn.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py
index 8eb388f2..dc9c62a8 100644
--- a/modules/deeplearning/srcnn.py
+++ b/modules/deeplearning/srcnn.py
@@ -362,15 +362,17 @@ class SRCNN:
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 = 0.1
- 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, kernel_initializer='he_uniform', padding=padding)(conv_b)
--
GitLab