diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py
index bb43831b960cc76345b5221b9edb525f8e727ae9..29686951c3c6a41921d6073329b9d0e2a5099315 100644
--- a/modules/deeplearning/cloud_opd_fcn_abi.py
+++ b/modules/deeplearning/cloud_opd_fcn_abi.py
@@ -359,6 +359,7 @@ class SRCNN:
         label = label[:, y_64, x_64]
         cld_prob = cld_prob[:, y_64, x_64]
         cat_cf = get_label_data_5cat(cld_prob)
+        cat_cf = np.pad(cat_cf, pad_width=[(0, 0), (1, 1), (1, 1)])
         data_norm.append(cat_cf)
         data = np.stack(data_norm, axis=3)
         data = data.astype(np.float32)