diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index 9ca728128ccdc562085f9c492d2db763d762c4a3..61702b4bdd885bb21f6c131aca66e0c2d83a1573 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)