Check if provided model is weights only and work around

brainglobe · Oct 8, 2024 · e3eabd8 · e3eabd8
1 parent 594b5c0
commit e3eabd8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/cellfinder/core/classify/classify.py b/cellfinder/core/classify/classify.py
@@ -68,6 +68,14 @@ def main(
         workers=workers,
     )
 
+    if trained_model and trained_model.suffix == ".h5":
+        print(
+            "Weights provided in place of the model, "
+            "loading weights into default model"
+        )
+        model_weights = trained_model
+        trained_model = None
+
     model = get_model(
         existing_model=trained_model,
         model_weights=model_weights,

diff --git a/debug.py b/debug.py