Changed weights PATH when loading and disabled OSX library

danflop · Sep 16, 2021 · 6300aa4 · 6300aa4
1 parent fa4764c
commit 6300aa4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pulmonary_embolism.py b/pulmonary_embolism.py
@@ -14,7 +14,7 @@
 import shutil
 import matplotlib as mpl
 
-mpl.use('TkAgg')
+#mpl.use('TkAgg')
 import matplotlib.pyplot as plt
 
 # Root directory of the project
@@ -358,7 +358,7 @@ def get_ax(rows=1, cols=1, size=8):
         MODEL_PATH = args.model
         print("Loading weights from ", MODEL_PATH)
 
-        model.load_weights(COCO_MODEL_PATH, by_name=True,
+        model.load_weights(MODEL_PATH, by_name=True,
                            exclude=["mrcnn_class_logits", "mrcnn_bbox_fc",
                                     "mrcnn_bbox", "mrcnn_mask"])
         print("finish")