20 import SimpleITK
as sitk
25 def command_iteration(method):
27 f
"{method.GetOptimizerIteration():3} "
28 + f
"= {method.GetMetricValue():10.5f}"
30 print(
"\t#: ", len(method.GetOptimizerPosition()))
33 def command_multi_iteration(method):
34 print(
"--------- Resolution Changing ---------")
41 "<fixedImageFilter> <movingImageFile>",
42 "<outputTransformFile>",
50 transformDomainMeshSize = [10] * moving.GetDimension()
53 print(
"Initial Parameters:")
54 print(tx.GetParameters())
57 R.SetMetricAsMattesMutualInformation(50)
58 R.SetOptimizerAsGradientDescentLineSearch(
59 5.0, 100, convergenceMinimumValue=1e-4, convergenceWindowSize=5
61 R.SetOptimizerScalesFromPhysicalShift()
62 R.SetInitialTransform(tx)
63 R.SetInterpolator(sitk.sitkLinear)
65 R.SetShrinkFactorsPerLevel([6, 2, 1])
66 R.SetSmoothingSigmasPerLevel([6, 2, 1])
68 R.AddCommand(sitk.sitkIterationEvent,
lambda: command_iteration(R))
70 sitk.sitkMultiResolutionIterationEvent,
lambda: command_multi_iteration(R)
73 outTx = R.Execute(fixed, moving)
77 print(f
"Optimizer stop condition: {R.GetOptimizerStopConditionDescription()}")
78 print(f
" Iteration: {R.GetOptimizerIteration()}")
79 print(f
" Metric value: {R.GetMetricValue()}")
83 if "SITK_NOSHOW" not in os.environ:
85 resampler.SetReferenceImage(fixed)
86 resampler.SetInterpolator(sitk.sitkLinear)
87 resampler.SetDefaultPixelValue(100)
88 resampler.SetTransform(outTx)
90 out = resampler.Execute(moving)
93 cimg =
sitk.Compose(simg1, simg2, simg1 // 2.0 + simg2 // 2.0)
94 sitk.Show(cimg,
"ImageRegistration1 Composition")