From 25083a84a1015a3d2435b760be34bb17a9fcfb08 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Mon, 24 Jul 2023 13:53:20 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/cloud_fraction_fcn_abi.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py index 8728acfc..9046bd58 100644 --- a/modules/deeplearning/cloud_fraction_fcn_abi.py +++ b/modules/deeplearning/cloud_fraction_fcn_abi.py @@ -73,7 +73,7 @@ print('data_params_full: ', data_params_full) print('label_param: ', label_param) KERNEL_SIZE = 3 -X_LEN = Y_LEN = 64 +X_LEN = Y_LEN = 128 if KERNEL_SIZE == 3: slc_x = slice(0, int(X_LEN/4) + 2) @@ -911,13 +911,14 @@ def run_restore_static(directory, ckpt_dir, out_file=None): nn = SRCNN() labels, preds, inputs = nn.run_restore(directory, ckpt_dir) if out_file is not None: + y_hi, x_hi = (Y_LEN // 4) + 1, (X_LEN // 4) + 1 np.save(out_file, [np.squeeze(labels), preds.argmax(axis=3), - denormalize(inputs[:, 1:17, 1:17, 0], 'temp_11_0um_nom', mean_std_dct), - denormalize(inputs[:, 1:17, 1:17, 1], 'refl_0_65um_nom', mean_std_dct), - denormalize(inputs[:, 1:17, 1:17, 2], 'refl_0_65um_nom', mean_std_dct), - inputs[:, 1:17, 1:17, 3], - inputs[:, 1:17, 1:17, 4]]) + denormalize(inputs[:, 1:y_hi, 1:x_hi, 0], 'temp_11_0um_nom', mean_std_dct), + denormalize(inputs[:, 1:y_hi, 1:x_hi, 1], 'refl_0_65um_nom', mean_std_dct), + denormalize(inputs[:, 1:y_hi, 1:x_hi, 2], 'refl_0_65um_nom', mean_std_dct), + inputs[:, 1:y_hi, 1:x_hi, 3], + inputs[:, 1:y_hi, 1:x_hi, 4]]) def run_evaluate_static(in_file, out_file, ckpt_dir): -- GitLab