OcmForegroundSceneLearner.h

Go to the documentation of this file.

#pragma once

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

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

#include <visualization/psm/ProbabilisticSceneModelVisualization.h>
#include <visualization/psm/ProbabilisticSceneVisualization.h>

#include <trainer/PSMTrainer.h>
#include <trainer/FullyMeshedTrainer.h>
#include <trainer/TopologyTreeTrainer.h>
#include <topology_creator/TopologyCreator.h>


// Local includes
#include "learner/foreground/ForegroundSceneLearner.h"

#include "learner/foreground/ocm/SceneObjectLearner.h"

#include "learner/foreground/ocm/ocm/OcmSceneObjectLearner.h"

#include <learner/foreground/ocm/combinatorial_optimization/CombinatorialTrainer.h>


namespace ProbabilisticSceneRecognition {
  
  class OcmForegroundSceneLearner : public ForegroundSceneLearner {    
  public:

    OcmForegroundSceneLearner(const ISM::ObjectSetPtr pExample);
    
    ~OcmForegroundSceneLearner();
    
    void initializeVisualizer(boost::shared_ptr<Visualization::ProbabilisticSceneModelVisualization> mSuperior);
    
    void save(boost::property_tree::ptree& pPt);
    
    void learn();
    
  private:
    
    std::vector<boost::shared_ptr<SceneObjectLearner> > mSceneObjectLearners;
    
    boost::shared_ptr<Visualization::ProbabilisticSceneVisualization> mVisualizer;
  };
}