CombinatorialTrainer.cpp

Go to the documentation of this file.

#include "learner/foreground/ocm/combinatorial_optimization/CombinatorialTrainer.h"

namespace ProbabilisticSceneRecognition {

  CombinatorialTrainer::CombinatorialTrainer(std::vector<boost::shared_ptr<SceneObjectLearner>> pLearners,
                                             std::vector<std::string> pObjectTypes)
      : SceneModel::AbstractTrainer()
  {
    std::cout << "Learning tree with combinatorial optimization." << std::endl;

    // Initilaize a source that translates evidence messages.
    examplesListSource = boost::shared_ptr<SceneModel::ExamplesListSource>(new SceneModel::ExamplesListSource());
    source = examplesListSource;

    // Initialize the generator for building the tree using combinatorial optimization.
    boost::shared_ptr<CombinatorialGraphGenerator> gen(new CombinatorialGraphGenerator(pLearners, pObjectTypes));
    generator = gen;

    std::cout << "Combinatorial optimization prepared." << std::endl;
  }

  CombinatorialTrainer::~CombinatorialTrainer()
  { }

  void CombinatorialTrainer::addSceneGraphMessages(std::vector<ISM::ObjectSetPtr> pMessages)
  {
      examplesListSource->addSceneGraphMessage(pMessages);
  }

}