|
228 | 228 | "from brainles_preprocessing.brain_extraction import HDBetExtractor\n", |
229 | 229 | "from brainles_preprocessing.modality import Modality\n", |
230 | 230 | "from brainles_preprocessing.preprocessor import Preprocessor\n", |
231 | | - "from brainles_preprocessing.registration import NiftyRegRegistrator\n", |
| 231 | + "from brainles_preprocessing.registration import (\n", |
| 232 | + " ANTsRegistrator,\n", |
| 233 | + " NiftyRegRegistrator,\n", |
| 234 | + " eRegRegistrator,\n", |
| 235 | + ")\n", |
| 236 | + "\n", |
| 237 | + "# ANTSRegistrator (default), NiftyRegRegistrator (Linux), eRegRegistrator (experimental)\n", |
232 | 238 | "\n", |
233 | 239 | "\n", |
234 | 240 | "def preprocess_exam_in_brats_style(inputDir: str) -> None:\n", |
|
319 | 325 | " preprocessor = Preprocessor(\n", |
320 | 326 | " center_modality=center,\n", |
321 | 327 | " moving_modalities=moving_modalities,\n", |
322 | | - " registrator=NiftyRegRegistrator(),\n", |
| 328 | + " registrator=ANTsRegistrator(),\n", |
323 | 329 | " brain_extractor=HDBetExtractor(),\n", |
324 | 330 | " # optional: we provide a temporary directory as a sandbox for the preprocessin\n", |
325 | 331 | " temp_folder=\"temporary_directory\",\n", |
|
0 commit comments