confusionmatrix.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <CVTypes.h>
#include <ecvMainAppInterface.h>
 
#include <QWidget>
#include <opencv2/core/mat.hpp>
#include <set>
 
namespace Ui {
class ConfusionMatrix;
}
 
class ConfusionMatrix : public QWidget {
    Q_OBJECT
 
public:
    enum metrics { PRECISION = 0, RECALL = 1, F1_SCORE = 2 };
 
    explicit ConfusionMatrix(const std::vector<ScalarType>& actual,
                             const std::vector<ScalarType>& predicted);
    ~ConfusionMatrix() override;
 
    void computePrecisionRecallF1Score(cv::Mat& matrix,
                                       cv::Mat& precisionRecallF1Score,
                                       cv::Mat& vec_TP_FN);
    float computeOverallAccuracy(cv::Mat& matrix);
    void compute(const std::vector<ScalarType>& actual,
                 const std::vector<ScalarType>& predicted);
    void setSessionRun(QString session, int run);
    bool save(QString filePath);
    float getOverallAccuracy();
 
private:
    std::set<ScalarType> classes;
    int nbClasses;
    Ui::ConfusionMatrix* ui;
    cv::Mat confusionMatrix;
    cv::Mat precisionRecallF1Score;
    std::vector<ScalarType> class_numbers;
    float m_overallAccuracy;
};