From f7e7730578438ab1617808c5ecac0919f76664b1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 14 Mar 2023 12:00:23 -0500
Subject: [PATCH] snapshot...
---
modules/deeplearning/cnn_cld_frac_mod_res.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/modules/deeplearning/cnn_cld_frac_mod_res.py b/modules/deeplearning/cnn_cld_frac_mod_res.py
index aff1f2a5..703865d8 100644
--- a/modules/deeplearning/cnn_cld_frac_mod_res.py
+++ b/modules/deeplearning/cnn_cld_frac_mod_res.py
@@ -402,7 +402,10 @@ class SRCNN:
label = input_label
label = label.copy()
label = label[:, y_128, x_128]
- label = get_label_data(label)
+ if NumClasses == 5:
+ label = get_label_data_5cat(label)
+ else:
+ label = get_label_data(label)
if label_param != 'cloud_probability':
label = normalize(label, label_param, mean_std_dct)
--
GitLab