example

Signed-off-by: neuronflow <[email protected]>
BrainLesion · Mar 30, 2024 · 9d9dac2 · 9d9dac2
1 parent f18917b
commit 9d9dac2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/example/example_modality_centric_preprocessor.py b/example/example_modality_centric_preprocessor.py
@@ -6,7 +6,11 @@
 from brainles_preprocessing.brain_extraction import HDBetExtractor
 from brainles_preprocessing.modality import Modality
 from brainles_preprocessing.preprocessor import Preprocessor
-from brainles_preprocessing.registration import ANTsRegistrator, NiftyRegRegistrator
+from brainles_preprocessing.registration import (
+    ANTsRegistrator,
+    NiftyRegRegistrator,
+    eRegRegistrator,
+)
 
 
 def preprocess(inputDir):
@@ -102,7 +106,8 @@ def preprocess(inputDir):
             moving_modalities=moving_modalities,
             # choose the registration backend you want to use
             # registrator=NiftyRegRegistrator(),
-            registrator=ANTsRegistrator(),
+            # registrator=ANTsRegistrator(),
+            registrator=eRegRegistrator(),
             brain_extractor=HDBetExtractor(),
             temp_folder="temporary_directory",
             limit_cuda_visible_devices="0",