Browse Source

Update 'automatic_segmentation_program/registration.py'

Frederik Filip Stæger 4 years ago
parent
commit
cbc9ade558
1 changed files with 3 additions and 3 deletions
  1. 3 3
      automatic_segmentation_program/registration.py

+ 3 - 3
automatic_segmentation_program/registration.py

@@ -18,7 +18,7 @@ def full_registration(fixed, moving, output='full_trans.nii', transform_pre=''):
         initial_moving_transform_com=1,
         interpolation='BSpline',
         metric_weight=[0.5] * 6,
-        radius_or_number_of_bins=[4,32] * 3,
+        radius_or_number_of_bins=[4, 32] * 3,
         sampling_strategy=['Regular'] * 3,
         use_histogram_matching=True,
         winsorize_lower_quantile=0.05,
@@ -37,7 +37,7 @@ def full_registration(fixed, moving, output='full_trans.nii', transform_pre=''):
 
 
 # A faster rigid registration
-def rigid_registration(fixed, moving, output='affine_trans.nii', transform_pre=''):
+def rigid_registration(fixed, moving, output='rigid_trans.nii', transform_pre=''):
     ants.Registration(
         dimension=2,
         transforms=['Rigid'],
@@ -47,7 +47,7 @@ def rigid_registration(fixed, moving, output='affine_trans.nii', transform_pre='
         convergence_threshold=[1.e-6],
         number_of_iterations=[[1000, 1000, 1000]],
         shrink_factors=[[32, 16, 8]],
-        initial_moving_transform_com=0,
+        initial_moving_transform_com=1,
         interpolation='BSpline',
         metric_weight=[1],
         radius_or_number_of_bins=[32],