diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 6bb63bdfe210b87dc95880c872ae9a9c91a27710..ecc1844ec1a4607debe5ed79c0ad2d36f917ebb9 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -268,7 +268,8 @@ class SRCNN: tmp = np.where(np.isnan(tmp), 0.0, tmp) tmp = tmp[:, self.slc_y_m, self.slc_x_m] tmp = self.upsample(tmp) - tmp = smooth_2d(tmp) + if DO_SMOOTH: + tmp = smooth_2d(tmp) tmp = normalize(tmp, param, mean_std_dct) data_norm.append(tmp) @@ -286,7 +287,8 @@ class SRCNN: # tmp = np.where(np.isnan(tmp), 0.0, tmp) # tmp = tmp[:, self.slc_y_2, self.slc_x_2] # tmp = self.upsample(tmp) - # tmp = smooth_2d(tmp) + # if DO_SMOOTH: + # tmp = smooth_2d(tmp) # tmp = normalize(tmp, label_param, mean_std_dct) # data_norm.append(tmp) @@ -295,7 +297,8 @@ class SRCNN: tmp = np.where(np.isnan(tmp), 0.0, tmp) tmp = tmp[:, self.slc_y_2, self.slc_x_2] tmp = self.upsample(tmp) - tmp = smooth_2d(tmp) + if DO_SMOOTH: + tmp = smooth_2d(tmp) tmp = normalize(tmp, label_param, mean_std_dct) data_norm.append(tmp)