TrackingEvaluationModule.h

#ifndef TRACKINGEVALUATIONMODULE_H
#define TRACKINGEVALUATIONMODULE_H

#include <map>
#include <string>
#include <QImage>
#include "Datapool.h"
#include "ModuleInterface.h"
#include <QDir>
#include "src/json.h"

#include "opencv2/opencv.hpp"


class TrackingEvaluationModule: public ModuleInterface {
public:
    TrackingEvaluationModule(Datapool *i_data);
    ~TrackingEvaluationModule();

    //Set module configuration parameters
    bool setParameters(QDomNode& config);

    //Initialization after reading parameters
    bool init();

    //Function executed at each frame
    bool run();

    //update parameters at runtime.
    bool updateParameters();

    void writeToLog();
    void getCurrentResults(QImage *);

    private:
        int TP;
        int TN;
        int FP;
        int FN;
        long N;
        int NN;
        long accTP;
        long accTN;
        long accFP;
        long accFN;
        double mTP;
        double mTN;
        double mFP;
        double mFN;
        double sdTP;
        double sdTN;
        double sdFP;
        double sdFN;
        double mean_blob;
        bool first;
        QString OutputName;
        QString InputName;
        std::map<unsigned long, std::set<unsigned long> > associations;
        std::map<unsigned long, double> distances;
        std::map<unsigned long, int> lost;
        Groundtruth internal_gt;
        int m_framesToWait;
        int waitedFrames;
        int internalFrameNumber;
        bool m_setROI;
        int m_dx;
        int m_dy;
        int m_w;
        int m_h;
        int m_wscale;
        bool m_logFirstTime;

        std::map<QString, double> previousParameters;
        std::map<QString, double> bestParameters;
        double bestFScore;
};

#endif