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