TopologyEvaluator.h

Go to the documentation of this file.

#pragma once

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

#include "helper/PrintHelper.h"


namespace ProbabilisticSceneRecognition {

class TopologyEvaluator: public AbstractTopologyEvaluator {
public:

    TopologyEvaluator(std::vector<boost::shared_ptr<ISM::ObjectSet>> pExamplesList,
              std::vector<boost::shared_ptr<SceneObjectLearner>> pLearners, double pRecognitionThreshold);

    ~TopologyEvaluator();

    bool evaluate(boost::shared_ptr<SceneModel::Topology> pTopology, bool pFullyMeshed = false);

private:

    std::pair<double, double> recognize(boost::shared_ptr<TestSet> pEvidence, bool pFullyMeshed = false);

    double getProbability(boost::shared_ptr<TestSet> pEvidence);

    void update(boost::shared_ptr<SceneModel::TreeNode> pTree);

    void xmlOutput(boost::shared_ptr<SceneModel::Topology> pTopology);

    std::vector<boost::shared_ptr<ISM::ObjectSet>> mExamplesList;

    std::vector<boost::shared_ptr<SceneObjectLearner>> mLearners;

    unsigned int mRunNumber;

    PrintHelper mPrintHelper;

    std::ofstream mRuntimeLogger;

    ForegroundSceneContent mForegroundSceneContent;

    std::string mInferenceAlgorithm;

    std::string mXmlOutput;

    std::string mXmlFilePath;

    boost::property_tree::ptree mModel;

    bool mCreateRuntimeLog;

    // Visualization parameters:
    bool mVisualize;
    bool mTargetingHelp;
    std::string mFrameId;
    double mScaleFactor;
    double mSigmaMultiplier;

    boost::shared_ptr<Visualization::ProbabilisticSceneVisualization> mVisualizer;   
};

}