From 66ab8528e6a091ef1b8c592ae836b5ff186102f4 Mon Sep 17 00:00:00 2001 From: tomrink <rink@ssec.wisc.edu> Date: Wed, 8 Jun 2022 13:44:22 -0500 Subject: [PATCH] snapshot... --- modules/deeplearning/unet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py index 5bd6ebea..6e5d2486 100644 --- a/modules/deeplearning/unet.py +++ b/modules/deeplearning/unet.py @@ -199,8 +199,8 @@ class UNET: data = data.astype(np.float32) label = label.astype(np.float32) - data = scale(data, param, None) - label = scale(label, param, None) + data = normalize(data, param, mean_std_dct) + label = normalize(label, param, mean_std_dct) if is_training and DO_AUGMENT: data_ud = np.flip(data, axis=1) -- GitLab