From 900cf2a47627a5b8de2d5a2cca058c4e0e95dffb Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 7 Jun 2022 13:27:53 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/unet.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/modules/deeplearning/unet.py b/modules/deeplearning/unet.py
index 6d371e2a..1044e416 100644
--- a/modules/deeplearning/unet.py
+++ b/modules/deeplearning/unet.py
@@ -1,7 +1,7 @@
 import glob
 import tensorflow as tf
 from util.setup import logdir, modeldir, cachepath, now, ancillary_path, home_dir
-from util.util import EarlyStop, normalize, make_for_full_domain_predict
+from util.util import EarlyStop, normalize, make_for_full_domain_predict, scale
 
 import os, datetime
 import numpy as np
@@ -199,8 +199,8 @@ class UNET:
         data = data.astype(np.float32)
         label = label.astype(np.float32)
 
-        data = normalize(data, param, mean_std_dct)
-        label = normalize(label, param, mean_std_dct)
+        data = scale(data, param, None)
+        label = scale(label, param, None)
 
         if is_training and DO_AUGMENT:
             data_ud = np.flip(data, axis=1)
-- 
GitLab