20 if len(sys.argv) != 5:
22 "Usage: " + sys.argv[0] +
" <inputImage> <outputImage> "
23 "<numberOfIterations> <conductance>")
26 inputImage = sys.argv[1]
27 outputImage = sys.argv[2]
28 numberOfIterations = int(sys.argv[3])
29 conductance = float(sys.argv[4])
32 InputPixelType = itk.UC
33 InputImageType =
itk.Image[InputPixelType, Dimension]
34 OutputPixelType = itk.F
35 OutputImageType =
itk.Image[OutputPixelType, Dimension]
38 reader = ReaderType.New()
39 reader.SetFileName(inputImage)
42 castfilter = CastFilterType.New()
43 castfilter.SetInput(reader)
46 OutputImageType, OutputImageType]
47 gradientfilter = FilterType.New()
48 gradientfilter.SetInput(castfilter.GetOutput())
49 gradientfilter.SetNumberOfIterations(numberOfIterations)
50 gradientfilter.SetTimeStep(0.125)
51 gradientfilter.SetConductanceParameter(conductance)
54 writer = WriterType.New()
55 writer.SetFileName(outputImage)
56 writer.SetInput(gradientfilter.GetOutput())