20 import SimpleITK
as sitk
25 def command_iteration(method):
26 if method.GetOptimizerIteration() == 0:
27 print(
"Estimated Scales: ", method.GetOptimizerScales())
29 f
"{method.GetOptimizerIteration():3} "
30 + f
"= {method.GetMetricValue():7.5f} "
31 + f
": {method.GetOptimizerPosition()}"
39 "ImageRegistrationMethod3"
40 "<fixedImageFilter> <movingImageFile> <outputTransformFile>",
50 R.SetMetricAsCorrelation()
52 R.SetOptimizerAsRegularStepGradientDescent(
55 numberOfIterations=500,
56 gradientMagnitudeTolerance=1e-8,
58 R.SetOptimizerScalesFromIndexShift()
63 R.SetInitialTransform(tx)
65 R.SetInterpolator(sitk.sitkLinear)
67 R.AddCommand(sitk.sitkIterationEvent,
lambda: command_iteration(R))
69 outTx = R.Execute(fixed, moving)
73 print(f
"Optimizer stop condition: {R.GetOptimizerStopConditionDescription()}")
74 print(f
" Iteration: {R.GetOptimizerIteration()}")
75 print(f
" Metric value: {R.GetMetricValue()}")
80 resampler.SetReferenceImage(fixed)
81 resampler.SetInterpolator(sitk.sitkLinear)
82 resampler.SetDefaultPixelValue(100)
83 resampler.SetTransform(outTx)
85 out = resampler.Execute(moving)
89 cimg =
sitk.Compose(simg1, simg2, simg1 // 2.0 + simg2 // 2.0)
91 return {
"fixed": fixed,
96 if __name__ ==
"__main__":
97 return_dict = main(sys.argv)
98 if "SITK_NOSHOW" not in os.environ:
99 sitk.Show(return_dict[
"composition"],
"ImageRegistration3 Composition")