25 if len(sys.argv) != 5:
27 "Usage: " + sys.argv[0] +
" <inputImage> <outputImage> "
28 "<numberOfIterations> <conductance>")
31 inputImage = sys.argv[1]
32 outputImage = sys.argv[2]
33 numberOfIterations = int(sys.argv[3])
34 conductance = float(sys.argv[4])
37 InputPixelType = itk.UC
38 InputImageType =
itk.Image[InputPixelType, Dimension]
39 OutputPixelType = itk.F
40 OutputImageType =
itk.Image[OutputPixelType, Dimension]
43 reader = ReaderType.New()
44 reader.SetFileName(inputImage)
47 castfilter = CastFilterType.New()
48 castfilter.SetInput(reader)
51 OutputImageType, OutputImageType]
52 gradientfilter = FilterType.New()
53 gradientfilter.SetInput(castfilter.GetOutput())
54 gradientfilter.SetNumberOfIterations(numberOfIterations)
55 gradientfilter.SetTimeStep(0.125)
56 gradientfilter.SetConductanceParameter(conductance)
59 writer = WriterType.New()
60 writer.SetFileName(outputImage)
61 writer.SetInput(gradientfilter.GetOutput())