40 if len(sys.argv) != 5:
41 print(
'Usage: ' + sys.argv[0] +
42 ' <InputFileName> <OutputFileName> <Threshold> <Level>')
45 inputFileName = sys.argv[1]
46 outputFileName = sys.argv[2]
50 FloatPixelType = itk.ctype(
'float')
51 FloatImageType =
itk.Image[FloatPixelType, Dimension]
54 reader.SetFileName(inputFileName)
62 threshold = float(sys.argv[3])
63 level = float(sys.argv[4])
64 watershed.SetThreshold(threshold)
65 watershed.SetLevel(level)
67 LabeledImageType = type(watershed.GetOutput())
69 PixelType = itk.ctype(
'unsigned char')
71 RGBImageType =
itk.Image[RGBPixelType, Dimension]
73 ScalarToRGBColormapFilterType = \
75 colormapImageFilter = ScalarToRGBColormapFilterType.New()
76 colormapImageFilter.SetColormap(ScalarToRGBColormapFilterType.Jet)
77 colormapImageFilter.SetInput(watershed.GetOutput())
78 colormapImageFilter.Update()
81 writer = WriterType.New()
82 writer.SetFileName(outputFileName)
83 writer.SetInput(colormapImageFilter.GetOutput())