diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py index a1f5db22ac10acfd4a807da6754a561f026d416e..c88f7cc77fb86202705650cc37da9f6784fd2f05 100644 --- a/modules/deeplearning/cloud_opd_srcnn_viirs.py +++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py @@ -255,7 +255,8 @@ class SRCNN: tmp = np.where(np.isnan(tmp), 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 = upsample_nearest(tmp) # tmp = tmp[:, self.slc_y, self.slc_x] @@ -287,7 +288,8 @@ class SRCNN: tmp = np.where(np.isnan(tmp), 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 = upsample_nearest(tmp) # tmp = tmp[:, self.slc_y, self.slc_x]