learner_batch.cpp

Go to the documentation of this file.

// Global includes
#include <cstdlib>

// Package includes
#include <ros/ros.h>

#include <learner/pbd_computation_graph.hpp>

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

using namespace ProbabilisticSceneRecognition;

int main(int argc, char* argv[])
{
  ros::init(argc, argv, "js_probabilistic_scene_learner");
  
  // Declaration of the learner for the scene model.
  SceneLearningEngine* learner;

  // Check for wrong configurations and similar.
  try {
    learner = new SceneLearningEngine();

  } catch(std::exception& exception){
    std::cerr << exception.what() << std::endl;
    std::exit(1);
  }

  // Get all ISM::ObjectSets from databasefiles passed as ros parameters.
  learner->readLearnerInput();

  // Check for errors with the resulting scene model
  try {       
    // Calculate parameters of scene model based on scene graphs from rosbag input.
    learner->generateSceneModel();
    
    // Dump model to file and plot its distributions.
    learner->saveSceneModel();
  } catch(std::exception& exception) {
    std::cerr << exception.what() << std::endl;
  }
  
  // Get rid of learner object.
  delete learner;

  // Stating that program has run correctly.
  return EXIT_SUCCESS;
}