From 360795283b8a8b50ca1ead2f824c675d5b4dfb67 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 28 Aug 2023 11:50:38 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/cloud_opd_fcn_abi.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index 9ca72812..61702b4b 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -1,5 +1,6 @@ import tensorflow as tf +from deeplearning.cloud_fraction_fcn_abi import get_label_data_5cat # from util.plot_cm import confusion_matrix_values from util.augment import augment_image from util.setup_cloud_fraction import logdir, modeldir, now, ancillary_path @@ -249,6 +250,7 @@ class SRCNN: self.test_preds = [] self.test_probs = None self.test_input = [] + self.test_cat_cf = [] self.learningRateSchedule = None self.num_data_samples = None @@ -329,6 +331,9 @@ class SRCNN: label = input_label[:, label_idx_i, :, :] label = label[:, y_64, x_64] cld_prob = cld_prob[:, y_64, x_64] + if not is_training: + cat_cf = get_label_data_5cat(cld_prob) + self.test_cat_cf.append(cat_cf) label = get_cldy_frac_opd(cld_prob, label) label = scale(label, label_param, mean_std_dct) -- GitLab