1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
|
#
# Example on the use of the CurvatureAnisotropicDiffusionImageFilter
#
from InsightToolkit import *
from sys import argv
reader = itkImageFileReaderUS2_New()
writer = itkImageFileWriterUS2_New()
inputCast = itkCastImageFilterUS2F2_New()
outputCast = itkRescaleIntensityImageFilterF2US2_New()
filter = itkCurvatureAnisotropicDiffusionImageFilterF2F2_New()
inputCast.SetInput( reader.GetOutput() )
filter.SetInput( inputCast.GetOutput() )
outputCast.SetInput( filter.GetOutput() )
writer.SetInput( outputCast.GetOutput() )
reader.SetFileName( argv[1] )
writer.SetFileName( argv[2] )
outputCast.SetOutputMinimum( 0 )
outputCast.SetOutputMaximum( 65535 )
numberOfIterations = eval( argv[3] )
timeStep = eval( argv[4] )
conductance = eval( argv[5] )
filter.SetNumberOfIterations( numberOfIterations )
filter.SetTimeStep( timeStep )
filter.SetConductanceParameter( conductance )
writer.Update()
|