diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index ecc1844ec1a4607debe5ed79c0ad2d36f917ebb9..361a1b2e01b345d0870fbf673b7b03aa31b82436 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -819,14 +819,16 @@ class SRCNN: bt = np.expand_dims(bt, axis=0) # bt_us = upsample_static(bt, x_2, y_2, t, s, None, None) bt_us = self.upsample(bt) - bt_us = smooth_2d(bt_us) + if DO_SMOOTH: + bt_us = smooth_2d(bt_us) bt_us = normalize(bt_us, 'temp_11_0um_nom', mean_std_dct) refl = np.where(np.isnan(refl), 0, refl) refl = refl[self.slc_y_m, self.slc_x_m] refl = np.expand_dims(refl, axis=0) refl_us = self.upsample(refl) - refl_us = smooth_2d(refl_us) + if DO_SMOOTH: + refl_us = smooth_2d(refl_us) refl_us = normalize(refl_us, 'refl_0_65um_nom', mean_std_dct) cld_opd = np.where(np.isnan(cld_opd), 0, cld_opd) @@ -834,7 +836,8 @@ class SRCNN: cld_opd = np.expand_dims(cld_opd, axis=0) # cld_opd_us = upsample_static(cld_opd, x_2, y_2, t, s, None, None) cld_opd_us = self.upsample(cld_opd) - cld_opd_us = smooth_2d(cld_opd_us) + if DO_SMOOTH: + cld_opd_us = smooth_2d(cld_opd_us) cld_opd_us = normalize(cld_opd_us, label_param, mean_std_dct) refl_sub_lo = np.expand_dims(refl_sub_lo, axis=0)