Use weights stored in object #54

CameronBodine · Jan 11, 2023 · b3ea43c · b3ea43c
1 parent 2832ec6
commit b3ea43c
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)