WIP writing brainreg output directory

brainglobe · Dec 11, 2024 · 189f52b · 189f52b
1 parent b2733e7
commit 189f52b
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 10 deletions.
diff --git a/brainglobe_registration/elastix/register.py b/brainglobe_registration/elastix/register.py
@@ -115,10 +115,17 @@ def run_registration(
     )[..., ::-1]
 
     if output_directory:
-        # Remove the deformationField.tiff file generated by elastix
-        os.unlink(output_directory / "deformationField.tiff")
+        working_directory = output_directory
     else:
-        os.unlink("deformationField.tiff")
+        working_directory = Path.cwd()
+
+    # Cleanup files generated by elastix
+    os.remove(working_directory / "deformationField.tiff")
+    for i in range(len(parameter_lists)):
+        try:
+            os.remove(working_directory / f"result.{i}.tiff")
+        except FileNotFoundError:
+            continue
 
     return (
         np.asarray(result_image),

diff --git a/brainglobe_registration/parameters/ara_tools/affine.txt b/brainglobe_registration/parameters/ara_tools/affine.txt
@@ -4,9 +4,9 @@
 
 //ImageTypes
 (FixedInternalImagePixelType "float")
-(FixedImageDimension 3)
+(FixedImageDimension 2)
 (MovingInternalImagePixelType "float")
-(MovingImageDimension 3)
+(MovingImageDimension 2)
 
 //Components
 (Registration "MultiResolutionRegistration")

diff --git a/brainglobe_registration/parameters/ara_tools/bspline.txt b/brainglobe_registration/parameters/ara_tools/bspline.txt
@@ -2,9 +2,9 @@
 
 //ImageTypes
 (FixedInternalImagePixelType "float")
-(FixedImageDimension 3)
+(FixedImageDimension 2)
 (MovingInternalImagePixelType "float")
-(MovingImageDimension 3)
+(MovingImageDimension 2)
 
 //Components
 (Registration "MultiResolutionRegistration")

diff --git a/brainglobe_registration/parameters/elastix_default/affine.txt b/brainglobe_registration/parameters/elastix_default/affine.txt
@@ -21,4 +21,4 @@
 (ResultImageFormat "tiff")
 (Transform "AffineTransform")
 (WriteIterationInfo "false")
-(WriteResultImage "false")
+(WriteResultImage "true")
diff --git a/brainglobe_registration/parameters/elastix_default/bspline.txt b/brainglobe_registration/parameters/elastix_default/bspline.txt
@@ -24,4 +24,4 @@
 (ResultImageFormat "tiff")
 (Transform "BSplineTransform")
 (WriteIterationInfo "false")
-(WriteResultImage "false")
+(WriteResultImage "true")
diff --git a/brainglobe_registration/parameters/elastix_default/rigid.txt b/brainglobe_registration/parameters/elastix_default/rigid.txt
@@ -21,4 +21,4 @@
 (ResultImageFormat "tiff")
 (Transform "EulerTransform")
 (WriteIterationInfo "false")
-(WriteResultImage "false")
+(WriteResultImage "true")