GaussianMixtureModel.h

Go to the documentation of this file.

#pragma once

// Global includes
#include <vector>

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

#include <visualization/psm/ProbabilisticSecondarySceneObjectVisualization.h>

// Local includes
#include "learner/foreground/ocm/ocm/shape/GaussianKernel.h"

namespace ProbabilisticSceneRecognition {
 
  class GaussianMixtureModel {
  public:

    GaussianMixtureModel();
    
    ~GaussianMixtureModel();
    
    void addKernel(const PSMLearner::GaussianKernel& pKernel);
    
    void normalizeWeights();
    
    unsigned int getNumberOfKernels();
    
    void initializeVisualizer(boost::shared_ptr<Visualization::ProbabilisticSecondarySceneObjectVisualization> mSuperior);
    
    void save(boost::property_tree::ptree& pPt, std::string pNode);

    std::vector<PSMLearner::GaussianKernel> getKernels() const;

  private:
    
    std::vector<PSMLearner::GaussianKernel> mKernels;
  };
}