From 4feea2731bbc71ad8787d4f5962887fd7da951bf Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 10 Sep 2023 12:39:37 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_fcn_abi.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 1ff295c9..6f0e1648 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -91,6 +91,11 @@ if KERNEL_SIZE == 3:
slc_y = slice(0, int(Y_LEN/4) + 2)
x_64 = slice(4, X_LEN + 4)
y_64 = slice(4, Y_LEN + 4)
+elif KERNEL_SIZE == 1:
+ slc_x = slice(4, X_LEN + 4)
+ slc_y = slice(4, Y_LEN + 4)
+ x_64 = slice(4, X_LEN + 4)
+ y_64 = slice(4, Y_LEN + 4)
# ----------------------------------------
@@ -515,7 +520,7 @@ class SRCNN:
conv_b = build_residual_conv2d_block(conv_b, num_filters, 'Residual_Block_6', kernel_size=KERNEL_SIZE, scale=scale)
- conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
+ conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=KERNEL_SIZE, strides=1, activation=activation, kernel_initializer='he_uniform', padding=padding)(conv_b)
# conv = conv + conv_b
conv = conv_b
--
GitLab