From fbc47efca3f47b823eedeef30d2c412110f11553 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Sun, 26 Mar 2023 14:47:07 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cloud_opd_srcnn.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/modules/deeplearning/cloud_opd_srcnn.py b/modules/deeplearning/cloud_opd_srcnn.py
index f541c9c4..a6e7ffab 100644
--- a/modules/deeplearning/cloud_opd_srcnn.py
+++ b/modules/deeplearning/cloud_opd_srcnn.py
@@ -14,7 +14,7 @@ LOG_DEVICE_PLACEMENT = False
PROC_BATCH_SIZE = 4
PROC_BATCH_BUFFER_SIZE = 5000
-NumClasses = 5
+NumClasses = 2
if NumClasses == 2:
NumLogits = 1
else:
@@ -499,7 +499,6 @@ class SRCNN:
@tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
def train_step(self, inputs, labels):
- labels = tf.squeeze(labels, axis=[3])
with tf.GradientTape() as tape:
pred = self.model([inputs], training=True)
loss = self.loss(labels, pred)
@@ -519,7 +518,6 @@ class SRCNN:
@tf.function(input_signature=[tf.TensorSpec(None, tf.float32), tf.TensorSpec(None, tf.float32)])
def test_step(self, inputs, labels):
- labels = tf.squeeze(labels, axis=[3])
pred = self.model([inputs], training=False)
t_loss = self.loss(labels, pred)
--
GitLab