/
q3DMASCClassifier.h
99 lines (81 loc) · 3.11 KB
/
q3DMASCClassifier.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
#pragma once
//##########################################################################
//# #
//# CLOUDCOMPARE PLUGIN: q3DMASC #
//# #
//# This program is free software; you can redistribute it and/or modify #
//# it under the terms of the GNU General Public License as published by #
//# the Free Software Foundation; version 2 or later of the License. #
//# #
//# This program is distributed in the hope that it will be useful, #
//# but WITHOUT ANY WARRANTY; without even the implied warranty of #
//# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
//# GNU General Public License for more details. #
//# #
//# COPYRIGHT: Dimitri Lague / CNRS / UEB #
//# #
//##########################################################################
//Local
#include "Parameters.h"
#include "FeaturesInterface.h"
//Qt
#include <QString>
//CCLib
#include <ReferenceCloud.h>
//OpenCV
#include <opencv2/ml.hpp>
class QWidget;
class ccMainAppInterface;
class ConfusionMatrix;
class Train3DMASCDialog;
//! 3DMASC classifier
namespace masc
{
class Classifier
{
public:
//! Default constructor
Classifier();
//! Train the classifier
bool train( const ccPointCloud* cloud,
const RandomTreesParams& params,
const Feature::Source::Set& featureSources,
QString& errorMessage,
CCCoreLib::ReferenceCloud* trainSubset = nullptr,
ccMainAppInterface* app = nullptr,
QWidget* parentWidget = nullptr);
//! Classifier accuracy metrics
struct AccuracyMetrics
{
unsigned sampleCount = 0;
unsigned goodGuess = 0;
float ratio = 0.0f;
};
//! Evaluates the classifier
bool evaluate( const Feature::Source::Set& featureSources,
ccPointCloud* testCloud,
AccuracyMetrics& metrics,
QString& errorMessage,
Train3DMASCDialog& train3DMASCDialog,
CCCoreLib::ReferenceCloud* testSubset = nullptr,
QString outputSFName = QString(),
QWidget* parentWidget = nullptr,
ccMainAppInterface* app = nullptr);
//! Applies the classifier
bool classify( const Feature::Source::Set& featureSources,
ccPointCloud* cloud,
QString& errorMessage,
QWidget* parentWidget = nullptr,
ccMainAppInterface* app = nullptr);
//! Returns whether the classifier is valid or not
bool isValid() const;
//! Saves the classifier to file
bool toFile(QString filename, QWidget* parentWidget = nullptr) const;
//! Loads the classifier from file
bool fromFile(QString filename, QWidget* parentWidget = nullptr);
inline cv::Mat getVarImportance() const { return m_rtrees->getVarImportance(); }
protected:
//! Random trees (OpenCV)
cv::Ptr<cv::ml::RTrees> m_rtrees;
};
}; //namespace masc