diff --git a/modules/deeplearning/cloud_opd_srcnn_viirs.py b/modules/deeplearning/cloud_opd_srcnn_viirs.py index 5253386f85f18c8a80bb98ef1e57f1825db4a699..8e023668d0ca23bcbedc87819cd10575ef13a965 100644 --- a/modules/deeplearning/cloud_opd_srcnn_viirs.py +++ b/modules/deeplearning/cloud_opd_srcnn_viirs.py @@ -2,7 +2,7 @@ import glob import tensorflow as tf from util.setup import logdir, modeldir, now, ancillary_path -from util.util import EarlyStop, normalize, denormalize, scale, descale, get_grid_values_all, resample_2d_linear +from util.util import EarlyStop, normalize, denormalize, scale, descale, get_grid_values_all, resample_2d_linear, smooth_2d import os, datetime import numpy as np import pickle @@ -255,6 +255,7 @@ 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) # --------------- # tmp = upsample_nearest(tmp) # tmp = tmp[:, self.slc_y, self.slc_x] @@ -286,6 +287,7 @@ 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) # ---------------- # tmp = upsample_nearest(tmp) # tmp = tmp[:, self.slc_y, self.slc_x]