Function main

Examples/ImageRegistrationMethod2/ImageRegistrationMethod2.cxx:60–108

int
main(int argc, char * argv[])
{

  if (argc < 4)
  {
    std::cerr << "Usage: " << argv[0] << " <fixedImageFilter> <movingImageFile> <outputTransformFile>" << std::endl;
    return 1;
  }


  sitk::Image fixed = sitk::ReadImage(argv[1], sitk::sitkFloat32);
  fixed = sitk::Normalize(fixed);
  fixed = sitk::DiscreteGaussian(fixed, 2.0);

  sitk::Image moving = sitk::ReadImage(argv[2], sitk::sitkFloat32);
  moving = sitk::Normalize(moving);
  moving = sitk::DiscreteGaussian(moving, 2.0);


  sitk::ImageRegistrationMethod R;
  R.SetMetricAsJointHistogramMutualInformation();

  const double       learningRate = 1;
  const unsigned int numberOfIterations = 200;
  const double       convergenceMinimumValue = 1e-4;
  const unsigned int convergenceWindowSize = 5;
  R.SetOptimizerAsGradientDescentLineSearch(
    learningRate, numberOfIterations, convergenceMinimumValue, convergenceWindowSize);

  R.SetInitialTransform(sitk::TranslationTransform(fixed.GetDimension()));
  R.SetInterpolator(sitk::sitkLinear);

  IterationUpdate cmd(R);
  R.AddCommand(sitk::sitkIterationEvent, cmd);

  sitk::Transform outTx = R.Execute(fixed, moving);

  std::cout << "-------" << std::endl;
  std::cout << outTx.ToString() << std::endl;
  std::cout << "Optimizer stop condition: " << R.GetOptimizerStopConditionDescription() << std::endl;
  std::cout << " Iteration: " << R.GetOptimizerIteration() << std::endl;
  std::cout << " Metric value: " << R.GetMetricValue() << std::endl;

  sitk::WriteTransform(outTx, argv[3]);


  return 0;
}