Skip to content

Commit 18ec6d3

Browse files
committed
api: changed smoothing sigmas to accept float in antsRegistration
1 parent ef2b7c1 commit 18ec6d3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

nipype/interfaces/ants/registration.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,7 @@ class RegistrationInputSpec(ANTSCommandInputSpec):
283283
traits.Float())))
284284
# Convergence flags
285285
number_of_iterations = traits.List(traits.List(traits.Int()))
286-
smoothing_sigmas = traits.List(traits.List(traits.Int()), mandatory=True)
286+
smoothing_sigmas = traits.List(traits.List(traits.Float()), mandatory=True)
287287
sigma_units = traits.List(traits.Enum('mm', 'vox'),
288288
requires=['smoothing_sigmas'],
289289
desc="units for smoothing sigmas", mandatory=True)
@@ -370,15 +370,15 @@ class Registration(ANTSCommand):
370370
>>> reg3.cmdline
371371
'antsRegistration --collapse-linear-transforms-to-fixed-image-header 0 --collapse-output-transforms 0 --dimensionality 3 --initial-moving-transform [ trans.mat, 1 ] --interpolation Linear --output [ output_, output_warped_image.nii.gz ] --transform Affine[ 2.0 ] --metric Mattes[ fixed1.nii, moving1.nii, 1, 32, Random, 0.05 ] --convergence [ 1500x200, 1e-08, 20 ] --smoothing-sigmas 1x0vox --shrink-factors 2x1 --use-estimate-learning-rate-once 1 --use-histogram-matching 1 --transform SyN[ 0.25, 3.0, 0.0 ] --metric Mattes[ fixed1.nii, moving1.nii, 1, 32 ] --convergence [ 100x50x30, 1e-09, 20 ] --smoothing-sigmas 2x1x0vox --shrink-factors 3x2x1 --use-estimate-learning-rate-once 1 --use-histogram-matching 1 --winsorize-image-intensities [ 0.025, 0.975 ] --write-composite-transform 1'
372372
373-
# Test collapse transforms flag
373+
>>> # Test collapse transforms flag
374374
>>> reg4 = copy.deepcopy(reg)
375375
>>> reg4.inputs.collapse_output_transforms = True
376376
>>> outputs = reg4._list_outputs()
377377
>>> print outputs #doctest: +ELLIPSIS
378378
{'reverse_invert_flags': [True, False], 'inverse_composite_transform': ['.../nipype/testing/data/output_InverseComposite.h5'], 'warped_image': '.../nipype/testing/data/output_warped_image.nii.gz', 'inverse_warped_image': <undefined>, 'forward_invert_flags': [False, False], 'reverse_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat', '.../nipype/testing/data/output_1InverseWarp.nii.gz'], 'composite_transform': ['.../nipype/testing/data/output_Composite.h5'], 'forward_transforms': ['.../nipype/testing/data/output_0GenericAffine.mat', '.../nipype/testing/data/output_1Warp.nii.gz']}
379379
>>> reg4.aggregate_outputs() #doctest: +SKIP
380380
381-
# Test multiple metrics per stage
381+
>>> # Test multiple metrics per stage
382382
>>> reg5 = copy.deepcopy(reg)
383383
>>> reg5.inputs.metric = ['CC', ['CC', 'Mattes']]
384384
>>> reg5.inputs.metric_weight = [1, [.5]*2]

0 commit comments

Comments
 (0)