mirror of
https://github.com/dgirardeau/q3DMASC.git
synced 2026-08-29 08:34:48 +08:00
Daniel Girardeau-Montaut
77 lines
2.6 KiB
C++
77 lines
2.6 KiB
C++
#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;
|
|
|
|
//! 3DMASC classifier
|
|
namespace masc
|
|
{
|
|
class Classifier
|
|
{
|
|
public:
|
|
|
|
//! Default constructor
|
|
Classifier();
|
|
|
|
//! Train the classifier
|
|
bool train(const ccPointCloud* cloud, const RandomTreesParams& params, const Feature::Set& features, QString& errorMessage, CCLib::ReferenceCloud* trainSubset = 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::Set& features, CCLib::ReferenceCloud* testSubset, AccuracyMetrics& metrics, QString& errorMessage, QWidget* parentWidget = nullptr);
|
|
|
|
//! Applies the classifier
|
|
bool classify(const Feature::Set& features, ccPointCloud* cloud, QString& errorMessage, QWidget* parentWidget = 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);
|
|
|
|
protected:
|
|
|
|
//! Random trees (OpenCV)
|
|
cv::Ptr<cv::ml::RTrees> m_rtrees;
|
|
};
|
|
|
|
}; //namespace masc
|