ForegroundSceneLearner.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <string>
#include <vector>

// Package includes
#include <boost/shared_ptr.hpp>
#include <boost/property_tree/ptree.hpp>

#include <visualization/psm/ProbabilisticSceneModelVisualization.h>

// Local includes
#include "learner/SceneLearner.h"

namespace ProbabilisticSceneRecognition {
  
  class ForegroundSceneLearner : public SceneLearner {    
  public:

    ForegroundSceneLearner(const ISM::ObjectSetPtr pExample);
    
    virtual ~ForegroundSceneLearner();
    
    virtual void initializeVisualizer(boost::shared_ptr<Visualization::ProbabilisticSceneModelVisualization> mSuperior) = 0;
    
    virtual void save(boost::property_tree::ptree& pPt) = 0;
    
    virtual void learn() = 0;
    
    void setClusteringParameters(double pStaticBreakRatio, double pTogetherRatio, double pMaxAngleDeviation);
    
  protected:
    
    double mStaticBreakRatio, mTogetherRatio, mMaxAngleDeviation;
  };
}