40 parser = argparse.ArgumentParser(description=
"Segment With Watershed Image Filter.")
41 parser.add_argument(
"input_image")
42 parser.add_argument(
"output_image")
43 parser.add_argument(
"threshold", type=float)
44 parser.add_argument(
"level", type=float)
45 args = parser.parse_args()
50 FloatPixelType = itk.ctype(
"float")
51 FloatImageType =
itk.Image[FloatPixelType, Dimension]
54 reader.SetFileName(args.input_image)
60 threshold = args.threshold
62 watershed.SetThreshold(threshold)
63 watershed.SetLevel(level)
65 LabeledImageType = type(watershed.GetOutput())
67 PixelType = itk.ctype(
"unsigned char")
69 RGBImageType =
itk.Image[RGBPixelType, Dimension]
72 LabeledImageType, RGBImageType
74 colormapImageFilter = ScalarToRGBColormapFilterType.New()
75 colormapImageFilter.SetColormap(
76 itk.ScalarToRGBColormapImageFilterEnums.RGBColormapFilter_Jet
78 colormapImageFilter.SetInput(watershed.GetOutput())
79 colormapImageFilter.Update()
82 writer = WriterType.New()
83 writer.SetFileName(args.output_image)
84 writer.SetInput(colormapImageFilter.GetOutput())