20 parser = argparse.ArgumentParser(description=
"Compute Curvature Flow.")
21 parser.add_argument(
"input_image")
22 parser.add_argument(
"output_image")
23 parser.add_argument(
"number_of_iterations", type=int)
24 parser.add_argument(
"time_step", type=float)
25 args = parser.parse_args()
27 InputPixelType = itk.F
28 OutputPixelType = itk.UC
31 InputImageType =
itk.Image[InputPixelType, Dimension]
32 OutputImageType =
itk.Image[OutputPixelType, Dimension]
35 reader = ReaderType.New()
36 reader.SetFileName(args.input_image)
39 curvatureFlowFilter = FilterType.New()
41 curvatureFlowFilter.SetInput(reader.GetOutput())
42 curvatureFlowFilter.SetNumberOfIterations(args.number_of_iterations)
43 curvatureFlowFilter.SetTimeStep(args.time_step)
46 rescaler = RescaleFilterType.New()
47 rescaler.SetInput(curvatureFlowFilter.GetOutput())
52 rescaler.SetOutputMinimum(outputPixelTypeMinimum)
53 rescaler.SetOutputMaximum(outputPixelTypeMaximum)
56 writer = WriterType.New()
57 writer.SetFileName(args.output_image)
58 writer.SetInput(rescaler.GetOutput())