/usr/include/OTB-6.4/otbKNearestNeighborsMachineLearningModel.h is in libotb-dev 6.4.0+dfsg-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 124 125 126 127 128 129 130 131 132 | /*
* Copyright (C) 2005-2017 Centre National d'Etudes Spatiales (CNES)
*
* This file is part of Orfeo Toolbox
*
* https://www.orfeo-toolbox.org/
*
* 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
*
* 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.
*/
#ifndef otbKNearestNeighborsMachineLearningModel_h
#define otbKNearestNeighborsMachineLearningModel_h
#include "otbRequiresOpenCVCheck.h"
#include "itkLightObject.h"
#include "itkFixedArray.h"
#include "otbMachineLearningModel.h"
#ifdef OTB_OPENCV_3
#include "otbOpenCVUtils.h"
#else
class CvKNearest;
#endif
namespace otb
{
template <class TInputValue, class TTargetValue>
class ITK_EXPORT KNearestNeighborsMachineLearningModel
: public MachineLearningModel <TInputValue, TTargetValue>
{
public:
/** Standard class typedefs. */
typedef KNearestNeighborsMachineLearningModel Self;
typedef MachineLearningModel<TInputValue, TTargetValue> Superclass;
typedef itk::SmartPointer<Self> Pointer;
typedef itk::SmartPointer<const Self> ConstPointer;
typedef typename Superclass::InputValueType InputValueType;
typedef typename Superclass::InputSampleType InputSampleType;
typedef typename Superclass::InputListSampleType InputListSampleType;
typedef typename Superclass::TargetValueType TargetValueType;
typedef typename Superclass::TargetSampleType TargetSampleType;
typedef typename Superclass::TargetListSampleType TargetListSampleType;
typedef typename Superclass::ConfidenceValueType ConfidenceValueType;
/** Run-time type information (and related methods). */
itkNewMacro(Self);
itkTypeMacro(KNearestNeighborsMachineLearningModel, MachineLearningModel);
/** Setters/Getters to the number of neighbors to use
* Default is 32
* \see http://docs.opencv.org/modules/ml/doc/k_nearest_neighbors.html
*/
itkGetMacro(K, int);
itkSetMacro(K, int);
/** Decision rule once the KNN are found :
* [for classification]
* - KNN_VOTING : output value with maximum occurrences (for classification)
* [for regression]
* - KNN_MEAN : output mean value of neighbors
* - KNN_MEDIAN : output median value of neighbors
*/
enum {KNN_VOTING, KNN_MEAN, KNN_MEDIAN};
/** Setters/Getters to the decision rule */
itkGetMacro(DecisionRule, int);
itkSetMacro(DecisionRule, int);
/** Train the machine learning model */
void Train() ITK_OVERRIDE;
/** Save the model to file */
void Save(const std::string & filename, const std::string & name="") ITK_OVERRIDE;
/** Load the model from file */
void Load(const std::string & filename, const std::string & name="") ITK_OVERRIDE;
/**\name Classification model file compatibility tests */
//@{
/** Is the input model file readable and compatible with the corresponding classifier ? */
bool CanReadFile(const std::string &) ITK_OVERRIDE;
/** Is the input model file writable and compatible with the corresponding classifier ? */
bool CanWriteFile(const std::string &) ITK_OVERRIDE;
//@}
protected:
/** Constructor */
KNearestNeighborsMachineLearningModel();
/** Destructor */
~KNearestNeighborsMachineLearningModel() ITK_OVERRIDE;
/** Predict values using the model */
TargetSampleType DoPredict(const InputSampleType& input, ConfidenceValueType *quality=ITK_NULLPTR) const ITK_OVERRIDE;
/** PrintSelf method */
void PrintSelf(std::ostream& os, itk::Indent indent) const ITK_OVERRIDE;
private:
KNearestNeighborsMachineLearningModel(const Self &); //purposely not implemented
void operator =(const Self&); //purposely not implemented
#ifdef OTB_OPENCV_3
cv::Ptr<cv::ml::KNearest> m_KNearestModel;
#else
CvKNearest * m_KNearestModel;
#endif
int m_K;
int m_DecisionRule;
};
} // end namespace otb
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbKNearestNeighborsMachineLearningModel.txx"
#endif
#endif
|