save current images

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Feb 14, 2024 · 78becc0 · 78becc0
1 parent 94ac819
commit 78becc0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/brainles_preprocessing/modality.py b/brainles_preprocessing/modality.py
@@ -252,7 +252,7 @@ def extract_brain_region(
             self.current = atlas_bet_cm
         return atlas_mask_path
 
-    def save_image(
+    def save_current_image(
         self,
         output_path: str,
         normalization=False,

diff --git a/brainles_preprocessing/preprocessor.py b/brainles_preprocessing/preprocessor.py
@@ -194,12 +194,12 @@ def run(
         # now we save images that are not skullstripped
         for modality in self.all_modalities:
             if modality.raw_skull_output_path:
-                modality.save_image(
+                modality.save_current_image(
                     modality.raw_skull_output_path,
                     normalization=False,
                 )
             if modality.normalized_skull_output_path:
-                modality.save_image(
+                modality.save_current_image(
                     modality.normalized_skull_output_path,
                     normalization=True,
                 )