CombinatorialTrainer.h

Go to the documentation of this file.

#pragma once

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

#include <trainer/TreeNode.h>
#include <trainer/AbstractTrainer.h>
#include <trainer/source/ExamplesListSource.h>

// Local includes
#include "learner/foreground/ocm/combinatorial_optimization/CombinatorialGraphGenerator.h"

namespace ProbabilisticSceneRecognition {

class CombinatorialTrainer: public SceneModel::AbstractTrainer
{
public:


  CombinatorialTrainer(std::vector<boost::shared_ptr<SceneObjectLearner>> pLearners,
                                               std::vector<std::string> pObjectTypes);

  ~CombinatorialTrainer();


  void addSceneGraphMessages(std::vector<ISM::ObjectSetPtr> pMessages);

private:

  boost::shared_ptr<SceneModel::ExamplesListSource> examplesListSource;

};

}