File: otbSVMImageClassificationWithRuleFilter.h

package info (click to toggle)
otb 5.8.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 38,496 kB
  • ctags: 40,282
  • sloc: cpp: 306,573; ansic: 3,575; python: 450; sh: 214; perl: 74; java: 72; makefile: 70
file content (124 lines) | stat: -rw-r--r-- 4,643 bytes parent folder | download
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
/*=========================================================================

  Program:   ORFEO Toolbox
  Language:  C++
  Date:      $Date$
  Version:   $Revision$


  Copyright (c) Centre National d'Etudes Spatiales. All rights reserved.
  See OTBCopyright.txt for details.

  Copyright (c) Institut Telecom; Telecom bretagne. All rights reserved.
  See IMTCopyright.txt 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 otbSVMImageClassificationWithRuleFilter_h
#define otbSVMImageClassificationWithRuleFilter_h

#include "otbSVMImageClassificationFilter.h"
#include "otbVectorImage.h"

namespace otb {

/** \class SVMImageClassificationWithRuleFilter
 * \brief SVM Image Classification that yield distance to hyperplanes
 *
 * This class comes from SVMImageClassificationFilter that performs
 * SVM classification on image. Here, the class add in ThreadedGenerateData an other output
 * (a VectorImage) that gives the value for \f$ f(x) \f$ without the majority
 * vote decision on class belonging.
 *
 * \sa SVMImageClassificationFilter
 * \ingroup Streamed
 * \ingroup Threaded
 *
 * \ingroup OTBSVMLearning
 */
template <class TInputImage, class TOutputImage, class TMaskImage = TOutputImage>
class ITK_EXPORT SVMImageClassificationWithRuleFilter
  : public SVMImageClassificationFilter<TInputImage, TOutputImage, TMaskImage>
{
public:
  /** Standard typedefs */
  typedef SVMImageClassificationWithRuleFilter Self;
  typedef SVMImageClassificationFilter<TInputImage,
      TOutputImage, TMaskImage>              Superclass;
  typedef itk::SmartPointer<Self>       Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;

  /** Type macro */
  itkNewMacro(Self);

  /** Creation through object factory macro */
  itkTypeMacro(SVMImageClassificationWithRuleFilter, SVMImageClassificationFilter);

  typedef typename Superclass::InputImageType             InputImageType;
  typedef typename Superclass::InputImageConstPointerType InputImageConstPointerType;
  typedef typename Superclass::ValueType                  ValueType;

  typedef typename Superclass::MaskImageType             MaskImageType;
  typedef typename Superclass::MaskImageConstPointerType MaskImageConstPointerType;
  typedef typename Superclass::MaskImagePointerType      MaskImagePointerType;

  typedef typename Superclass::OutputImageType        OutputImageType;
  typedef typename Superclass::OutputImagePointerType OutputImagePointerType;
  typedef typename Superclass::OutputImageRegionType  OutputImageRegionType;
  typedef typename Superclass::LabelType              LabelType;

  typedef typename Superclass::ModelType          ModelType;
  typedef typename Superclass::ModelPointerType   ModelPointerType;
  typedef typename ModelType::NodeCacheType       NodeCacheType;
  typedef typename ModelType::MeasurementType     MeasurementType;
  typedef typename ModelType::DistancesVectorType DistanceValueType;
  typedef typename DistanceValueType::ValueType   RuleValueType;

  typedef VectorImage<RuleValueType, OutputImageType::ImageDimension>
  OutputRuleImageType;
  typedef typename OutputRuleImageType::Pointer    OutputRuleImagePointerType;
  typedef typename OutputRuleImageType::RegionType OutputRuleImageRegionType;
  typedef typename OutputRuleImageType::PixelType  RuleType;

  /** Access to the Rule image */
  itkGetObjectMacro(OutputRule, OutputRuleImageType);

protected:
  /** Constructor */
  SVMImageClassificationWithRuleFilter();
  /** Destructor */
  ~SVMImageClassificationWithRuleFilter() ITK_OVERRIDE {}

  /** GenerateOutputInformation
        * Set the number of bands of the output rule image.
        * Copy information from the input image if existing.
        **/
  void GenerateOutputInformation() ITK_OVERRIDE;

  /** AllocateOutputs
   * Output allocation redefinition for VectorImage (used in TOutputRuleImage)
   **/
  void AllocateOutputs() ITK_OVERRIDE;

  /** Threaded generate data */
  void ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) ITK_OVERRIDE;

private:
  SVMImageClassificationWithRuleFilter(const Self &); //purposely not implemented
  void operator =(const Self&); //purposely not implemented

  OutputRuleImagePointerType m_OutputRule;

}; // end of class

} // end of namespace otb

#ifndef OTB_MANUAL_INSTANTIATION
#include "otbSVMImageClassificationWithRuleFilter.txx"
#endif

#endif