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
|
import itk
from sys import argv
itk.auto_progress(2)
InputPixelType = itk.F
OutputPixelType = itk.F
InputImageType = itk.Image[InputPixelType, 2]
OutputImageType = itk.Image[OutputPixelType, 2]
reader = itk.ImageFileReader[InputImageType].New( FileName=argv[1] )
filter = itk.GradientAnisotropicDiffusionImageFilter[InputImageType, OutputImageType].New( reader,
NumberOfIterations=int(argv[3]),
TimeStep=float(argv[4]),
ConductanceParameter=float(argv[5]))
filter.Update();
WritePixelType = itk.UC
WriteImageType = itk.Image[WritePixelType, 2]
rescaler = itk.RescaleIntensityImageFilter[OutputImageType, WriteImageType].New( filter, OutputMinimum=0, OutputMaximum=255 )
writer = itk.ImageFileWriter[WriteImageType].New( rescaler, FileName=argv[2] )
writer.Update();
|
