`snapshot...`

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)