1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19import InsightToolkit as itk
20import sys
21
22reader = itk.itkImageFileReaderF2_New()
23reader.SetFileName(sys.argv[1])
24
25diffusion = itk.itkGradientAnisotropicDiffusionImageFilterF2F2_New()
26diffusion.SetInput(reader.GetOutput())
27diffusion.SetTimeStep(0.0625)
28diffusion.SetConductanceParameter(9.0)
29diffusion.SetNumberOfIterations(5)
30
31gradient = itk.itkGradientMagnitudeImageFilterF2F2_New()
32gradient.SetInput(diffusion.GetOutput())
33
34watershed = itk.itkWatershedImageFilterF2_New()
35watershed.SetInput(gradient.GetOutput())
36watershed.SetThreshold(0.01)
37watershed.SetLevel(0.2)
38
39writer = itk.itkImageFileWriterUL2_New()
40writer.SetFileName(sys.argv[2])
41writer.SetInput(watershed.GetOutput())
42writer.Update()