24 from distutils.version
import StrictVersion
as VS
26 print(
"ITK 4.8.0 is required (see example documentation).")
29 if len(sys.argv) != 3:
30 print(
"Usage: " + sys.argv[0] +
" <inputImage> <outputImage>")
33 inputImage = sys.argv[1]
34 outputImage = sys.argv[2]
38 ImageType =
itk.Image[PixelType, Dimension]
41 RGBImageType =
itk.Image[RGBPixelType, Dimension]
44 reader = ReaderType.New()
45 reader.SetFileName(inputImage)
47 ColormapType = itk.CustomColormapFunction[PixelType, RGBPixelType]
48 colormap = ColormapType.New()
50 random = itk.MersenneTwisterRandomVariateGenerator.New()
58 redChannel.append(random.GetUniformVariate(0.0, 1.0))
59 greenChannel.append(random.GetUniformVariate(0.0, 1.0))
60 blueChannel.append(random.GetUniformVariate(0.0, 1.0))
62 colormap.SetRedChannel(redChannel)
63 colormap.SetGreenChannel(greenChannel)
64 colormap.SetBlueChannel(blueChannel)
67 ImageType, RGBImageType]
68 colormapFilter1 = ColormapFilterType.New()
70 colormapFilter1.SetInput(reader.GetOutput())
71 colormapFilter1.SetColormap(colormap)
74 writer = WriterType.New()
75 writer.SetFileName(outputImage)
76 writer.SetInput(colormapFilter1.GetOutput())