forked from InsightSoftwareConsortium/ITK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
itkOtsuMultipleThresholdsImageFilter.txx
105 lines (86 loc) · 3.84 KB
/
itkOtsuMultipleThresholdsImageFilter.txx
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
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkOtsuMultipleThresholdsImageFilter.txx
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#ifndef _itkOtsuMultipleThresholdsImageFilter_txx
#define _itkOtsuMultipleThresholdsImageFilter_txx
#include "itkOtsuMultipleThresholdsImageFilter.h"
#include "itkThresholdLabelerImageFilter.h"
#include "itkProgressAccumulator.h"
#include "itkScalarImageToHistogramGenerator.h"
#include "itkOtsuMultipleThresholdsCalculator.h"
namespace itk {
template<class TInputImage, class TOutputImage>
OtsuMultipleThresholdsImageFilter<TInputImage, TOutputImage>
::OtsuMultipleThresholdsImageFilter()
{
m_NumberOfHistogramBins = 128;
m_NumberOfThresholds = 1;
m_LabelOffset = NumericTraits<OutputPixelType>::Zero;
m_Thresholds.clear();
}
template<class TInputImage, class TOutputImage>
void
OtsuMultipleThresholdsImageFilter<TInputImage, TOutputImage>
::GenerateData()
{
typename ProgressAccumulator::Pointer progress = ProgressAccumulator::New();
progress->SetMiniPipelineFilter(this);
// Create a histogram of the image intensities
typedef itk::Statistics::ScalarImageToHistogramGenerator< TInputImage > HistogramGeneratorType;
typename HistogramGeneratorType::Pointer histogramGenerator = HistogramGeneratorType::New();
histogramGenerator->SetInput( this->GetInput() );
histogramGenerator->SetNumberOfBins( m_NumberOfHistogramBins );
histogramGenerator->Compute();
// Compute the multiple Otsu Thresholds for the input image
typedef typename HistogramGeneratorType::HistogramType HistogramType;
typedef OtsuMultipleThresholdsCalculator< HistogramType > OtsuType;
typename OtsuType::Pointer otsuThresholdCalculator = OtsuType::New();
otsuThresholdCalculator->SetInputHistogram( histogramGenerator->GetOutput() );
otsuThresholdCalculator->SetNumberOfThresholds( m_NumberOfThresholds );
otsuThresholdCalculator->Update();
m_Thresholds = otsuThresholdCalculator->GetOutput();
typename ThresholdLabelerImageFilter<TInputImage,TOutputImage>::Pointer threshold =
ThresholdLabelerImageFilter<TInputImage,TOutputImage>::New();;
progress->RegisterInternalFilter(threshold,.5f);
threshold->GraftOutput (this->GetOutput());
threshold->SetInput (this->GetInput());
threshold->SetThresholds( m_Thresholds );
threshold->SetLabelOffset( m_LabelOffset );
threshold->Update();
this->GraftOutput(threshold->GetOutput());
}
template<class TInputImage, class TOutputImage>
void
OtsuMultipleThresholdsImageFilter<TInputImage, TOutputImage>
::GenerateInputRequestedRegion()
{
const_cast<TInputImage *>(this->GetInput())->SetRequestedRegionToLargestPossibleRegion();
}
template<class TInputImage, class TOutputImage>
void
OtsuMultipleThresholdsImageFilter<TInputImage,TOutputImage>
::PrintSelf(std::ostream& os, Indent indent) const
{
Superclass::PrintSelf(os,indent);
os << indent << "NumberOfHistogramBins: " << m_NumberOfHistogramBins << std::endl;
os << indent << "NumberOfThresholds: " << m_NumberOfThresholds << std::endl;
os << indent << "LabelOffset: " << m_LabelOffset << std::endl;;
os << indent << "Thresholds: " << std::endl;
for (unsigned long j=0; j<m_Thresholds.size(); j++)
{
os << "\tThreshold #" << j << ": "
<< static_cast<typename NumericTraits<InputPixelType>::PrintType>(m_Thresholds[j])
<< std::endl;
}
}
}// end namespace itk
#endif