ITK  5.2.0
Insight Toolkit
Examples/Segmentation/VoronoiSegmentation.py
1 # ==========================================================================
2 #
3 # Copyright NumFOCUS
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0.txt
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16 #
17 # ==========================================================================*/
18 
19 #
20 # Example on the use of the VoronoiSegmentationImageFilter.
21 #
22 
23 from InsightToolkit import *
24 
25 from sys import argv
26 
27 
28 readerInput = itkImageFileReaderUC2_New()
29 readerPrior = itkImageFileReaderUC2_New()
30 
31 readerInput.SetFileName(argv[1])
32 readerPrior.SetFileName(argv[2])
33 
34 readerInput.Update()
35 readerPrior.Update()
36 
37 filter = itkVoronoiSegmentationImageFilterUC2UC2UC2_New()
38 
39 filter.SetInput(readerInput.GetOutput())
40 filter.TakeAPrior(readerPrior.GetOutput())
41 
42 filter.SetMeanPercentError(eval(argv[4]))
43 filter.SetSTDPercentError(eval(argv[5]))
44 
45 writer = itkImageFileWriterUC2_New()
46 writer.SetFileName(argv[3])
47 writer.SetInput(filter.GetOutput())
48 
49 writer.Update()