18 #ifndef itkKdTreeBasedKmeansEstimator_h
19 #define itkKdTreeBasedKmeansEstimator_h
22 #include <unordered_map>
73 template <
typename TKdTree>
129 itkSetMacro(MaximumIteration,
int);
130 itkGetConstMacro(MaximumIteration,
int);
135 itkSetMacro(CentroidPositionChangesThreshold,
double);
136 itkGetConstMacro(CentroidPositionChangesThreshold,
double);
140 SetKdTree(TKdTree * tree);
148 itkGetConstMacro(CurrentIteration,
int);
149 itkGetConstMacro(CentroidPositionChanges,
double);
160 itkSetMacro(UseClusterLabels,
bool);
161 itkGetConstMacro(UseClusterLabels,
bool);
168 PrintSelf(std::ostream & os,
Indent indent)
const override;
196 return static_cast<int>(m_Candidates.size());
205 m_Candidates.resize(centroids.size());
206 for (
unsigned int i = 0; i < centroids.size(); ++i)
213 m_Candidates[i] = candidate;
224 centroids.resize(this->
Size());
225 for (i = 0; i < static_cast<unsigned int>(this->
Size()); ++i)
227 centroids[i] = m_Candidates[i].Centroid;
238 for (i = 0; i < static_cast<unsigned int>(this->
Size()); ++i)
240 if (m_Candidates[i].
Size > 0)
242 for (j = 0; j < m_MeasurementVectorSize; ++j)
244 m_Candidates[i].Centroid[j] =
245 m_Candidates[i].WeightedCentroid[j] / static_cast<double>(m_Candidates[i].
Size);
268 GetSumOfSquaredPositionChanges(InternalParametersType & previous, InternalParametersType & current);
273 GetClosestCandidate(ParameterType & measurements, std::vector<int> & validIndexes);
277 IsFarther(ParameterType & pointA,
278 ParameterType & pointB,
279 MeasurementVectorType & lowerBound,
280 MeasurementVectorType & upperBound);
285 Filter(KdTreeNodeType * node,
286 std::vector<int> validIndexes,
287 MeasurementVectorType & lowerBound,
288 MeasurementVectorType & upperBound);
292 CopyParameters(InternalParametersType & source, InternalParametersType & target);
296 CopyParameters(ParametersType & source, InternalParametersType & target);
300 CopyParameters(InternalParametersType & source, ParametersType & target);
304 GetPoint(ParameterType &
point, MeasurementVectorType measurements);
307 PrintPoint(ParameterType &
point);
311 int m_CurrentIteration{ 0 };
314 int m_MaximumIteration{ 100 };
317 double m_CentroidPositionChanges{ 0.0 };
321 double m_CentroidPositionChangesThreshold{ 0.0 };
336 bool m_UseClusterLabels{
false };
337 bool m_GenerateClusterLabels{
false };
345 #ifndef ITK_MANUAL_INSTANTIATION
346 # include "itkKdTreeBasedKmeansEstimator.hxx"