Merge pull request #55 from CameronBodine/main

Use weights stored in object #54
CameronBodine · Jan 24, 2023 · 826d81f · 826d81f
2 parents fb98e8e + b3ea43c
commit 826d81f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/class_portstarObj.py b/src/class_portstarObj.py
@@ -480,7 +480,7 @@ def _initModel(self,
                         )
 
         try:
-            model = tf.keras.models.load_model(weights)
+            model = tf.keras.models.load_model(self.weights)
         except:
             model.compile(optimizer = 'adam', loss = dice_coef_loss, metrics = [mean_iou, dice_coef])
             model.load_weights(self.weights)