diff --git a/modules/deeplearning/cloud_opd_srcnn_abi.py b/modules/deeplearning/cloud_opd_srcnn_abi.py index cf8955ebaecc874f2985e059fd43580942c8f9e1..17d0e3abc9b3b57165844173d64708abda446dac 100644 --- a/modules/deeplearning/cloud_opd_srcnn_abi.py +++ b/modules/deeplearning/cloud_opd_srcnn_abi.py @@ -3,7 +3,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 @@ -750,6 +750,7 @@ class SRCNN: bt = bt[slc_y, slc_x] bt = np.expand_dims(bt, axis=0) bt_us = upsample_static(bt, x_2, y_2, t, s, None, None) + bt_us = smooth_2d(bt_us) bt_us = normalize(bt_us, 'temp_11_0um_nom', mean_std_dct) print('BT done') @@ -771,6 +772,7 @@ class SRCNN: cld_opd = cld_opd[slc_y, slc_x] 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 = smooth_2d(cld_opd_us) cld_opd_us = normalize(cld_opd_us, label_param, mean_std_dct) print('OPD done') t1 = time.time()