20 from distutils.version
import StrictVersion
as VS
22 print(
"ITK 4.8.0 is required (see example documentation).")
25 if len(sys.argv) != 3:
26 print(
"Usage: " + sys.argv[0] +
" <inputImage> <outputImage>")
29 inputImage = sys.argv[1]
30 outputImage = sys.argv[2]
34 ImageType =
itk.Image[PixelType, Dimension]
37 RGBImageType =
itk.Image[RGBPixelType, Dimension]
40 reader = ReaderType.New()
41 reader.SetFileName(inputImage)
43 ColormapType = itk.CustomColormapFunction[PixelType, RGBPixelType]
44 colormap = ColormapType.New()
46 random = itk.MersenneTwisterRandomVariateGenerator.New()
54 redChannel.append(random.GetUniformVariate(0.0, 1.0))
55 greenChannel.append(random.GetUniformVariate(0.0, 1.0))
56 blueChannel.append(random.GetUniformVariate(0.0, 1.0))
58 colormap.SetRedChannel(redChannel)
59 colormap.SetGreenChannel(greenChannel)
60 colormap.SetBlueChannel(blueChannel)
63 ImageType, RGBImageType]
64 colormapFilter1 = ColormapFilterType.New()
66 colormapFilter1.SetInput(reader.GetOutput())
67 colormapFilter1.SetColormap(colormap)
70 writer = WriterType.New()
71 writer.SetFileName(outputImage)
72 writer.SetInput(colormapFilter1.GetOutput())