otbSVMMarginSampler.h

Go to the documentation of this file.

/*=========================================================================

  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 __otbSVMMarginSampler_h
#define __otbSVMMarginSampler_h

#include "itkSampleClassifier.h"
#include "itkVectorImage.h"

namespace otb
{

template< class TSample, class TModel >
class ITK_EXPORT SVMMarginSampler :
      public itk::Statistics::SampleClassifier< TSample >
{
public:
  typedef SVMMarginSampler Self;
  typedef itk::Statistics::SampleClassifier< TSample > Superclass;
  typedef itk::SmartPointer< Self > Pointer;
  typedef itk::SmartPointer<const Self> ConstPointer;

  itkTypeMacro(SVMMarginSampler, itk::Statistics::SampleClassifier);
  itkNewMacro(Self);

  typedef itk::Statistics::MembershipSample< TSample > OutputType;

  typedef typename TSample::MeasurementType MeasurementType;
  typedef typename TSample::MeasurementVectorType MeasurementVectorType;
  typedef std::pair<unsigned int, double> IndexAndDistanceType;
  typedef std::vector<IndexAndDistanceType> IndexAndDistanceVectorType;
  typedef std::vector<unsigned int>         IndexVectorType;


  static bool Compare(const IndexAndDistanceType&  l1, const IndexAndDistanceType & l2)
    {
      bool resp = l2.second > l1.second;
      return resp;
    }

   
  typedef typename Superclass::MembershipFunctionPointerVector
  MembershipFunctionPointerVector;

  OutputType* GetOutput();
  void SetOutput(OutputType* output);

  typedef TModel                                     SVMModelType;
  typedef typename SVMModelType::Pointer             SVMModelPointer;
  typedef typename SVMModelType::DistancesVectorType DistancesVectorType;

  itkSetMacro(NumberOfCandidates, unsigned int);
  itkGetMacro(NumberOfCandidates, unsigned int);

  itkSetObjectMacro(Model, SVMModelType);

  itkGetObjectMacro(Model, SVMModelType);

  void Update();

  const IndexVectorType & GetMarginSamples()
  {
    return m_MarginSamples;
  }

protected:
  SVMMarginSampler();
  virtual ~SVMMarginSampler() {}
  void PrintSelf(std::ostream& os, itk::Indent indent) const;

  void GenerateData();
  virtual void DoMarginSampling();

private:
  typename OutputType::Pointer m_Output;

  SVMModelPointer m_Model;

  unsigned int m_NumberOfCandidates;

  IndexVectorType m_MarginSamples;

}; // end of class


} // end of namespace otb


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

#endif