20 import SimpleITK
as sitk
25 def command_iteration(method):
26 print(f
"{method.GetOptimizerIteration():3} " + f
"= {method.GetMetricValue():10.5f}")
27 print(
"\t#: ", len(method.GetOptimizerPosition()))
30 def command_multi_iteration(method):
31 print(
"--------- Resolution Changing ---------")
38 "<fixedImageFilter> <movingImageFile>",
39 "<outputTransformFile>",
47 transformDomainMeshSize = [10] * moving.GetDimension()
50 print(
"Initial Parameters:")
51 print(tx.GetParameters())
54 R.SetMetricAsMattesMutualInformation(50)
55 R.SetOptimizerAsGradientDescentLineSearch(
56 5.0, 100, convergenceMinimumValue=1e-4, convergenceWindowSize=5
58 R.SetOptimizerScalesFromPhysicalShift()
59 R.SetInitialTransform(tx)
60 R.SetInterpolator(sitk.sitkLinear)
62 R.SetShrinkFactorsPerLevel([6, 2, 1])
63 R.SetSmoothingSigmasPerLevel([6, 2, 1])
65 R.AddCommand(sitk.sitkIterationEvent,
lambda: command_iteration(R))
66 R.AddCommand(sitk.sitkMultiResolutionIterationEvent,
lambda: command_multi_iteration(R))
68 outTx = R.Execute(fixed, moving)
72 print(f
"Optimizer stop condition: {R.GetOptimizerStopConditionDescription()}")
73 print(f
" Iteration: {R.GetOptimizerIteration()}")
74 print(f
" Metric value: {R.GetMetricValue()}")
78 if "SITK_NOSHOW" not in os.environ:
80 resampler.SetReferenceImage(fixed)
81 resampler.SetInterpolator(sitk.sitkLinear)
82 resampler.SetDefaultPixelValue(100)
83 resampler.SetTransform(outTx)
85 out = resampler.Execute(moving)
88 cimg =
sitk.Compose(simg1, simg2, simg1 // 2.0 + simg2 // 2.0)
89 sitk.Show(cimg,
"ImageRegistration1 Composition")