AbstractTopologyEvaluator.h

Go to the documentation of this file.

#pragma once

#include <trainer/TopologyTreeTrainer.h>
#include <topology_creator/Topology.h>

#include <boost/property_tree/xml_parser.hpp>

#include <ISM/common_type/ObjectSet.hpp>

#include "learner/foreground/ForegroundSceneLearner.h"
#include "learner/foreground/ocm/SceneObjectLearner.h"
#include "learner/foreground/ocm/combinatorial_optimization/TestSet.h"

#include "inference/model/foreground/ForegroundSceneContent.h"

namespace ProbabilisticSceneRecognition {

class AbstractTopologyEvaluator {
public:

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

    void setValidTestSets(const std::vector<boost::shared_ptr<TestSet>>& pValidTestSets);

    std::vector<boost::shared_ptr<TestSet>> getValidTestSets();

    void setInvalidTestSets(const std::vector<boost::shared_ptr<TestSet>>& pInvalidTestSets);

    std::vector<boost::shared_ptr<TestSet>> getInvalidTestSets();

    void setRecognitionThreshold(double pRecognitionThreshold);

    double getRecognitionThreshold();

    void eraseValidTestSet(unsigned int pIndex);

    void eraseInvalidTestSet(unsigned int pIndex);

protected:

    std::vector<boost::shared_ptr<TestSet>> mValidTestSets;
    std::vector<boost::shared_ptr<TestSet>> mInvalidTestSets;


    double mRecognitionThreshold;

};

}