Merge branch 'master' of github.com:FNNDSC/pl-emerald

FNNDSC · Mar 7, 2024 · ccc034d · ccc034d
2 parents 62965d5 + f5a2f1d
commit ccc034d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/emerald/emerald.py b/emerald/emerald.py
@@ -135,7 +135,7 @@ def emerald(model: Unet, input_path: str, mask_path: Optional[Path], brain_paths
 
     #save result
     if mask_path:
-        save(res, str(mask_path), hdr)
+        save(res.astype(np.float32), str(mask_path), hdr)
 
     if brain_paths:
         # for whatever reason, img.shape=(38, 256, 256, 1).