From 6300aa451ff64b877b998a430d1bf6475cf54d64 Mon Sep 17 00:00:00 2001
From: unknown <dfloplo@gmail.com>
Date: Thu, 16 Sep 2021 14:34:03 -0500
Subject: [PATCH] Changed weights PATH when loading and disabled OSX library

---
 pulmonary_embolism.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pulmonary_embolism.py b/pulmonary_embolism.py
index fe463c4..5dc4314 100644
--- 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")