segmentationEvaluationModule.h

#ifndef SEGMENTATIONEVALUATIONMODULE_H
#define SEGMENTATIONEVALUATIONMODULE_H

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

#include "opencv2/opencv.hpp"


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

    //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 RoiFileName;
        bool m_setROI;
        int m_dx;
        int m_dy;
        int m_w;
        int m_h;
        int m_wscale;
        bool m_logFirstTime;
        bool firstTime;
        cv::Mat RoiPixelLevel;

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

#endif