BoostableClassifier.h

Go to the documentation of this file.

/* XBoost: Ada-Boost and Friends on Haar/ICF/HOG Features, Library and ToolBox
 *
 * Copyright (c) 2008-2014 Paolo Medici <medici@ce.unipr.it>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

#ifndef _BOOSTABLE_CLASSIFIER_H
#define _BOOSTABLE_CLASSIFIER_H

#include <string>
#include <vector>
#include "Types.h"


template<class WeakClassifier>
struct BoostableClassifier : public WeakClassifier {
  
  float alpha;
  
public:

  typedef WeakClassifier WeakClassifierType;
  
  static const ClassifierType Type = RealClassifier;
  
public:
  
  BoostableClassifier() : alpha(0.0f) { }
  BoostableClassifier(const WeakClassifier & weak, float a) : WeakClassifier(weak), alpha(a) { }

// NOTE build and convert costructor with 1 parameter overlapps with previous constructor
  
  template<class R, class P1, class P2>
  explicit BoostableClassifier(const R & src, P1 p1, P2 p2) : WeakClassifier(static_cast<const typename R::WeakClassifierType &>(src), p1, p2), alpha(src.alpha)  {  }

  template<class R, class P0, class P1, class P2>
  explicit BoostableClassifier(const R & src, P0 p0, P1 p1, P2 p2) : WeakClassifier(static_cast<const typename R::WeakClassifierType &>(src), p0, p1, p2), alpha(src.alpha)  {  }
  
  // TODO: boost- here is not correct. Boost is BoostClassifier. This class associate only weight to a Discrete Weak Classifier
  static std::string signature() { return std::string("boost-") + WeakClassifier::signature(); }

   template<class DataType>
  inline float operator()(const DataType data) const;
  
  template<class DataType, class TParam1>
  inline float operator()(const DataType data, TParam1 stride) const;
  
  inline float getAlpha() const { return alpha; }
  
};

template<class WeakClassifier>
inline std::istream & operator >> (std::istream & in, BoostableClassifier<WeakClassifier> & s);

template<class WeakClassifier>
inline std::ostream & operator << (std::ostream & out, const BoostableClassifier<WeakClassifier> & s);



template<class WeakClassifier>
template<class DataType, class TParam1>
float BoostableClassifier<WeakClassifier>::operator()(const DataType data, TParam1 stride) const
  {
    return WeakClassifier::operator()(data, stride) * alpha; 
  }

template<class WeakClassifier>
template<class DataType>
float BoostableClassifier<WeakClassifier>::operator()(const DataType data) const
  {
    return WeakClassifier::operator()(data) * alpha;
  }

template<class WeakClassifier>
inline std::istream & operator >> (std::istream & in, BoostableClassifier<WeakClassifier> & s)
  {
  in >> s.alpha >> static_cast<WeakClassifier&>(s);
  return in;
  }
  
template<class WeakClassifier>
inline std::ostream & operator << (std::ostream & out, const BoostableClassifier<WeakClassifier> & s)
  {
  out << s.alpha << ' '  << static_cast<const WeakClassifier&>(s);
  return out;
  }

  
#endif