diff --git a/modules/deeplearning/cloud_opd_fcn_abi.py b/modules/deeplearning/cloud_opd_fcn_abi.py index ddf0c8eee93e3913cd418ce3675f7597fb8a4cc9..b13c348e175ad0d25b6205d7dfce4b91168cdd4f 100644 --- a/modules/deeplearning/cloud_opd_fcn_abi.py +++ b/modules/deeplearning/cloud_opd_fcn_abi.py @@ -383,7 +383,8 @@ 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)]) + if KERNEL_SIZE != 1: + 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)