ITK  5.4.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# https://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
23from InsightToolkit import *
24
25from sys import argv
26
27
28readerInput = itkImageFileReaderUC2_New()
29readerPrior = itkImageFileReaderUC2_New()
30
31readerInput.SetFileName(argv[1])
32readerPrior.SetFileName(argv[2])
33
34readerInput.Update()
35readerPrior.Update()
36
37filter = itkVoronoiSegmentationImageFilterUC2UC2UC2_New()
38
39filter.SetInput(readerInput.GetOutput())
40filter.TakeAPrior(readerPrior.GetOutput())
41
42filter.SetMeanPercentError(eval(argv[4]))
43filter.SetSTDPercentError(eval(argv[5]))
44
45writer = itkImageFileWriterUC2_New()
46writer.SetFileName(argv[3])
47writer.SetInput(filter.GetOutput())
48
49writer.Update()