24 if len(sys.argv) != 5:
25 print(
"Usage: " + sys.argv[0] +
" <inputImage> <outputImage> "
26 "<numberOfIterations> <timeStep>")
29 inputImage = sys.argv[1]
30 outputImage = sys.argv[2]
31 numberOfIterations = int(sys.argv[3])
32 timeStep = float(sys.argv[4])
34 InputPixelType = itk.F
35 OutputPixelType = itk.UC
38 InputImageType =
itk.Image[InputPixelType, Dimension]
39 OutputImageType =
itk.Image[OutputPixelType, Dimension]
42 reader = ReaderType.New()
43 reader.SetFileName(inputImage)
46 InputImageType, InputImageType]
47 curvatureFlowFilter = FilterType.New()
49 curvatureFlowFilter.SetInput(reader.GetOutput())
50 curvatureFlowFilter.SetNumberOfIterations(numberOfIterations)
51 curvatureFlowFilter.SetTimeStep(timeStep)
54 InputImageType, OutputImageType]
55 rescaler = RescaleFilterType.New()
56 rescaler.SetInput(curvatureFlowFilter.GetOutput())
61 rescaler.SetOutputMinimum(outputPixelTypeMinimum)
62 rescaler.SetOutputMaximum(outputPixelTypeMaximum)
65 writer = WriterType.New()
66 writer.SetFileName(outputImage)
67 writer.SetInput(rescaler.GetOutput())