OcmSceneObjectLearner.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <vector>

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

#include <trainer/TreeNode.h>

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

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

#include "learner/foreground/ocm/ocm/OcmModel.h"
#include "learner/foreground/ocm/ocm/TermLearner.h"

#include "learner/foreground/ocm/ocm/shape/ShapeTermLearner.h"

#include "learner/foreground/ocm/ocm/appearance/AppearanceTermLearner.h"

#include "learner/foreground/ocm/ocm/occlusion/OcclusionTermLearner.h"

namespace ProbabilisticSceneRecognition {
  
  class OcmSceneObjectLearner : public SceneObjectLearner {
  public:

    OcmSceneObjectLearner(std::string pSceneObjectType, std::string pSceneName = "");
    
    ~OcmSceneObjectLearner();
    
    void initializeVisualizer(boost::shared_ptr<Visualization::ProbabilisticSceneVisualization> mSuperior);
    
    void save(boost::property_tree::ptree& pPt);
    
   void learn(std::vector<ISM::ObjectSetPtr> pExamplesList, boost::shared_ptr<SceneModel::TreeNode> pTree);
   
  private:
    
    boost::shared_ptr<OcmModel> mOcmModel;
    
    std::vector<boost::shared_ptr<TermLearner> > mTermLearners;
    
    boost::shared_ptr<Visualization::ProbabilisticPrimarySceneObjectVisualization> mVisualizer;
  };
}