OTB  6.1.0
Orfeo Toolbox
otbMRFEnergyGaussianClassification.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2005-2017 Centre National d'Etudes Spatiales (CNES)
3  *
4  * This file is part of Orfeo Toolbox
5  *
6  * https://www.orfeo-toolbox.org/
7  *
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  * http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  */
20 
21 #ifndef otbMRFEnergyGaussianClassification_h
22 #define otbMRFEnergyGaussianClassification_h
23 
24 #include "otbMRFEnergy.h"
25 #include "otbMath.h"
26 
27 namespace otb
28 {
48 template<class TInput1, class TInput2>
49 class ITK_EXPORT MRFEnergyGaussianClassification : public MRFEnergy<TInput1, TInput2>
50 {
51 public:
56 
57  typedef TInput1 InputImageType;
58  typedef TInput2 LabelledImageType;
59  typedef typename InputImageType::PixelType InputImagePixelType;
60  typedef typename LabelledImageType::PixelType LabelledImagePixelType;
62 
63  itkNewMacro(Self);
64 
66 
67  void SetNumberOfParameters(const unsigned int nParameters) ITK_OVERRIDE
68  {
69  Superclass::SetNumberOfParameters(nParameters);
70  this->m_Parameters.SetSize(nParameters);
71  this->Modified();
72  }
73 
74  double GetSingleValue(const InputImagePixelType& value1, const LabelledImagePixelType& value2) ITK_OVERRIDE
75  {
76  if ((unsigned int) value2 >= this->GetNumberOfParameters() / 2)
77  {
78  itkExceptionMacro(<< "Number of parameters does not correspond to number of classes");
79  }
80  double val1 = static_cast<double>(value1);
81 
82  double result = vnl_math_sqr(val1 - this->m_Parameters[2 * static_cast<int>(value2)])
83  / (2 * vnl_math_sqr(this->m_Parameters[2 * static_cast<int>(value2) + 1]))
84  + vcl_log(vcl_sqrt(CONST_2PI) * this->m_Parameters[2 * static_cast<int>(value2) + 1]);
85 
86  return static_cast<double>(result);
87  }
88 
89 protected:
90  // The constructor and destructor.
93 
94 };
95 }
96 
97 #endif
void SetNumberOfParameters(const unsigned int nParameters) ITK_OVERRIDE
This is the implementation of the Gaussian model for Markov classification.
const double CONST_2PI
Definition: otbMath.h:54
InputImageType::PixelType InputImagePixelType
Definition: otbMRFEnergy.h:53
double GetSingleValue(const InputImagePixelType &value1, const LabelledImagePixelType &value2) ITK_OVERRIDE
This is the base class for energy function used in the MRF framework.
Definition: otbMRFEnergy.h:43
LabelledImageType::PixelType LabelledImagePixelType
Definition: otbMRFEnergy.h:54