-
-
Notifications
You must be signed in to change notification settings - Fork 663
/
ResampleImageFilter.py
123 lines (93 loc) · 3.26 KB
/
ResampleImageFilter.py
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
#==========================================================================
#
# Copyright Insight Software Consortium
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0.txt
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
#==========================================================================*/
# INPUTS: {BrainProtonDensitySlice.png}
# OUTPUTS: {ResampleImageFilterOutput1.png}
# 0
# INPUTS: {BrainProtonDensitySlice.png}
# OUTPUTS: {ResampleImageFilterOutput2.png}
# 1
# INPUTS: {BrainProtonDensitySlice.png}
# OUTPUTS: {ResampleImageFilterOutput3.png}
# 2
# INPUTS: {BrainProtonDensitySlice.png}
# OUTPUTS: {ResampleImageFilterOutput4.png}
# 3
from __future__ import print_function
import itk
from sys import argv, stderr, exit
itk.auto_progress(2)
if(len(argv) < 3):
print((
"Missing Parameters \n Usage: ResampleImageFilter.py inputImageFile "
"outputImageFile [exampleAction={0,1,2,3}]"), file=stderr)
exit(1)
exampleAction = 0
if(len(argv) >= 4):
exampleAction = int(argv[3])
Dimension = 2
InputPixelType = itk.UC
OutputPixelType = itk.UC
InputImageType = itk.Image[InputPixelType, Dimension]
OutputImageType = itk.Image[OutputPixelType, Dimension]
ReaderType = itk.ImageFileReader[InputImageType]
CastType = itk.CastImageFilter[InputImageType, OutputImageType]
WriterType = itk.ImageFileWriter[OutputImageType]
reader = ReaderType.New()
cast = CastType.New()
writer = WriterType.New()
reader.SetFileName(argv[1])
writer.SetFileName(argv[2])
FilterType = itk.ResampleImageFilter[InputImageType, InputImageType]
filter = FilterType.New()
TransformType = itk.AffineTransform[itk.D, Dimension]
transform = TransformType.New()
filter.SetTransform(transform)
InterpolatorType = itk.NearestNeighborInterpolateImageFunction[
InputImageType, itk.D]
interpolator = InterpolatorType.New()
filter.SetInterpolator(interpolator)
filter.SetDefaultPixelValue(0)
# spacing = itk.Vector[itk.D, Dimension]()
# spacing.SetElement(0, 1.0)
# spacing.SetElement(1, 1.0)
# filter.SetOutputSpacing( spacing )
filter.SetOutputSpacing(1.0)
# origin = itk.Point[itk.D, Dimension]()
# origin.SetElement(0, 0.0)
# origin.SetElement(1, 0.0)
# filter.SetOutputOrigin( origin )
filter.SetOutputOrigin([0.0, 0.0])
# size = itk.Size[Dimension]()
# size.SetElement(0, 300)
# size.SetElement(1, 300)
# filter.SetSize( size )
filter.SetSize(300)
filter.SetInput(reader.GetOutput())
cast.SetInput(filter.GetOutput())
writer.SetInput(cast.GetOutput())
writer.Update()
# translation = itk.Vector[itk.D, Dimension]()
# translation.SetElement(0, -30)
# translation.SetElement(1, -50)
# transform.Translate( translation, 0 )
transform.Translate([-30, -50], False)
if(exampleAction == 1):
writer.Update()
filter.SetDefaultPixelValue(100)
if(exampleAction == 2):
writer.Update()