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
|
/*=========================================================================
Program: ORFEO Toolbox
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Centre National d'Etudes Spatiales. All rights reserved.
See OTBCopyright.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 otbSOMImageClassificationFilter_h
#define otbSOMImageClassificationFilter_h
#include "otbSOMClassifier.h"
#include "itkInPlaceImageFilter.h"
#include "itkListSample.h"
namespace otb
{
/** \class SOMImageClassificationFilter
* \brief This filter performs the classification of a VectorImage using a given SOM map.
*
* This filter is streamed and threaded, allowing to classify huge images. Because the
* internal sample type has to be an itk::FixedArray, one must specify at compilation time
* the maximum sample dimension. It is up to the user to specify a MaxSampleDimension sufficiently
* high to integrate all its features. This filter internally use one SVMClassifier per thread.
*
* \sa SVMClassifier
* \ingroup Streamed
* \ingroup Threaded
*
* \ingroup OTBSOM
*/
template <class TInputImage, class TOutputImage, class TSOMMap, class TMaskImage = TOutputImage>
class ITK_EXPORT SOMImageClassificationFilter
: public itk::InPlaceImageFilter<TInputImage, TOutputImage>
{
public:
/** Standard typedefs */
typedef SOMImageClassificationFilter Self;
typedef itk::InPlaceImageFilter<TInputImage, TOutputImage> Superclass;
typedef itk::SmartPointer<Self> Pointer;
typedef itk::SmartPointer<const Self> ConstPointer;
/** Type macro */
itkNewMacro(Self);
/** Creation through object factory macro */
itkTypeMacro(SOMImageClassificationFilter, InPlaceImageFilter);
typedef TInputImage InputImageType;
typedef typename InputImageType::ConstPointer InputImageConstPointerType;
typedef typename InputImageType::InternalPixelType ValueType;
typedef TMaskImage MaskImageType;
typedef typename MaskImageType::ConstPointer MaskImageConstPointerType;
typedef typename MaskImageType::Pointer MaskImagePointerType;
typedef TOutputImage OutputImageType;
typedef typename OutputImageType::Pointer OutputImagePointerType;
typedef typename OutputImageType::RegionType OutputImageRegionType;
typedef typename OutputImageType::PixelType LabelType;
typedef TSOMMap SOMMapType;
typedef typename SOMMapType::Pointer SOMMapPointerType;
typedef typename SOMMapType::PixelType SampleType;
typedef itk::Statistics::ListSample<SampleType> ListSampleType;
typedef typename ListSampleType::Pointer ListSamplePointerType;
typedef otb::SOMClassifier<ListSampleType, SOMMapType, LabelType> ClassifierType;
typedef typename ClassifierType::Pointer ClassifierPointerType;
/** Set/Get the svm model */
itkSetObjectMacro(Map, SOMMapType);
itkGetObjectMacro(Map, SOMMapType);
/** Set/Get the default label */
itkSetMacro(DefaultLabel, LabelType);
itkGetMacro(DefaultLabel, LabelType);
/**
* If set, only pixels within the mask will be classified.
* \param mask The input mask.
*/
void SetInputMask(const MaskImageType * mask);
/**
* Get the input mask.
* \return The mask.
*/
const MaskImageType * GetInputMask(void);
protected:
/** Constructor */
SOMImageClassificationFilter();
/** Destructor */
~SOMImageClassificationFilter() ITK_OVERRIDE {}
/** Threaded generate data */
void ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread, itk::ThreadIdType threadId) ITK_OVERRIDE;
/** Before threaded generate data */
void BeforeThreadedGenerateData() ITK_OVERRIDE;
/**PrintSelf method */
void PrintSelf(std::ostream& os, itk::Indent indent) const ITK_OVERRIDE;
private:
SOMImageClassificationFilter(const Self &); //purposely not implemented
void operator =(const Self&); //purposely not implemented
/** The SOM model used for classification */
SOMMapPointerType m_Map;
/** Default label for invalid pixels (when using a mask) */
LabelType m_DefaultLabel;
};
} // End namespace otb
#ifndef OTB_MANUAL_INSTANTIATION
#include "otbSOMImageClassificationFilter.txx"
#endif
#endif
|
