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