From 39caf54c05c54d76829cef8381f006bf97585b70 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 1 Aug 2022 13:44:00 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/espcn.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py
index 2f02c9a2..06abdbfd 100644
--- a/modules/deeplearning/espcn.py
+++ b/modules/deeplearning/espcn.py
@@ -419,14 +419,14 @@ class ESPCN:
if NOISE_TRAINING:
conv = tf.keras.layers.GaussianNoise(stddev=NOISE_STDDEV)(conv)
- conv = tf.keras.layers.Conv2D(num_filters, kernel_size=5, strides=1, padding=padding, activation=activation)(conv)
- conv = tf.keras.layers.BatchNormalization()(conv)
- print(conv.shape)
-
conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
conv = tf.keras.layers.BatchNormalization()(conv)
print(conv.shape)
+ # conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=activation)(conv)
+ # conv = tf.keras.layers.BatchNormalization()(conv)
+ # print(conv.shape)
+
conv = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, padding=padding, activation=None)(conv)
conv = tf.keras.layers.BatchNormalization()(conv)
print(conv.shape)
--
GitLab