Fix for header/bounding box bug

mist-medical · May 13, 2024 · 508187b · 508187b
1 parent ae4a7b8
commit 508187b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/mist/inference/main_inference.py b/mist/inference/main_inference.py
@@ -97,6 +97,9 @@ def back_to_original_space(pred, og_ants_img, config, fg_bbox):
     if fg_bbox is not None:
         pred = decrop_from_fg(pred, fg_bbox)
 
+    # FIX: Copy header from original image onto the prediction so they match
+    pred = og_ants_img.new_image_like(pred.numpy())
+
     return pred
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "mist-medical"
-version = "0.4.6-alpha"
+version = "0.4.7-alpha"
 requires-python = ">= 3.8"
 description = "MIST is a simple, fully automated framework for 3D medical imaging segmentation."
 readme = "README.md"