diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index 962aaabf1d28bcad250ac1e2036825d2535f0a81..146cd75d7cbfd91005343974632442c57deefdb6 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -31,7 +31,7 @@ NOISE_TRAINING = False NOISE_STDDEV = 0.01 DO_AUGMENT = True -DO_SMOOTH = True +DO_SMOOTH = False SIGMA = 1.0 DO_ZERO_OUT = False @@ -270,8 +270,7 @@ 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) - # if DO_SMOOTH: - if False: + if DO_SMOOTH: tmp = smooth_2d(tmp) tmp = normalize(tmp, param, mean_std_dct) data_norm.append(tmp) @@ -876,8 +875,7 @@ 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) - # if DO_SMOOTH: - if False: + if DO_SMOOTH: bt_us = smooth_2d(bt_us) bt_us = normalize(bt_us, 'temp_11_0um_nom', mean_std_dct) @@ -887,8 +885,7 @@ class SRCNN: refl = np.expand_dims(refl, axis=0) # refl_us = self.upsample(refl) refl_us = refl - # if DO_SMOOTH: - if False: + if DO_SMOOTH: refl_us = smooth_2d(refl) refl_us = normalize(refl_us, 'refl_0_65um_nom', mean_std_dct)