CLI works well on Windows

andreped · Sep 10, 2021 · 40c3fd9 · 40c3fd9
1 parent d2b1e41
commit 40c3fd9
Showing 1 changed file with 2 additions and 11 deletions.
diff --git a/livermask/livermask.py b/livermask/livermask.py
@@ -14,12 +14,8 @@
 import tensorflow as tf
 
 
-# due to this: https://github.com/tensorflow/tensorflow/issues/35029
-os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
-
-
-# mute some warnings
-warnings.filterwarnings('ignore', '.*output shape of zoom.*')
+os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'  # due to this: https://github.com/tensorflow/tensorflow/issues/35029
+warnings.filterwarnings('ignore', '.*output shape of zoom.*')  # mute some warnings
 
 
 def intensity_normalization(volume, intensity_clipping_range):
@@ -32,23 +28,18 @@ def intensity_normalization(volume, intensity_clipping_range):
     max_val = np.amax(result)
     if (max_val - min_val) != 0:
         result = (result - min_val) / (max_val - min_val)
-
     return result
 
 def post_process(pred):
     return pred
 
 def get_model(output):
     url = "https://drive.google.com/uc?id=12or5Q79at2BtLgQ7IaglNGPFGRlEgEHc"
-    # output = "./model.h5"
     md5 = "ef5a6dfb794b39bea03f5496a9a49d4d"
     gdown.cached_download(url, output, md5=md5) #, postprocess=gdown.extractall)
 
 def func(path, output, cpu):
-
     cwd = "/".join(os.path.realpath(__file__).replace("\\", "/").split("/")[:-1]) + "/"
-
-    print("Current cwd:", cwd)
     name = cwd + "model.h5"
 
     # get model