saberbud/q3DMASC
1
0
Fork 0
mirror of https://github.com/dgirardeau/q3DMASC.git synced 2026-08-29 16:40:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

New training dialog + selection of the most important features

Browse Source
This commit is contained in:
Daniel Girardeau-Montaut
2019-01-20 22:12:20 +01:00
parent 4ec7252b80
commit b5e5597e2d
7 changed files with 419 additions and 111 deletions
Download Patch File Download Diff File Expand all files Collapse all files
q3DMASCClassifier.cpp
+19 -1
View File
@@ -29,6 +29,9 @@
//qCC_io
#include <LASFields.h>
//qCC_plugins
#include <ccMainAppInterface.h>
//Qt
#include <QCoreApplication>
#include <QProgressDialog>
@@ -360,6 +363,7 @@ bool Classifier::train( const ccPointCloud* cloud,
const Feature::Set& features,
QString& errorMessage,
CCLib::ReferenceCloud* trainSubset/*=nullptr*/,
ccMainAppInterface* app/*=nullptr*/,
QWidget* parentWidget/*=nullptr*/)
{
if (features.empty())
@@ -397,7 +401,10 @@ bool Classifier::train( const ccPointCloud* cloud,
int sampleCount = static_cast<int>(trainSubset ? trainSubset->size() : cloud->size());
int attributesPerSample = static_cast<int>(features.size());
ccLog::Print(QString("[3DMASC] Training data: %1 samples with %2 feature(s)").arg(sampleCount).arg(attributesPerSample));
if (app)
{
app->dispToConsole(QString("[3DMASC] Training data: %1 samples with %2 feature(s)").arg(sampleCount).arg(attributesPerSample));
}
cv::Mat training_data, train_labels;
try
@@ -530,6 +537,17 @@ bool Classifier::train( const ccPointCloud* cloud,
return false;
}
if (app)
{
cv::Mat mat = m_rtrees->getVarImportance();
app->dispToConsole(QString("Var importance size = %1 x %2").arg(mat.rows).arg(mat.cols));
assert(attributesPerSample == mat.rows);
for (int i = 0; i < mat.rows; ++i)
{
app->dispToConsole(QString("Feature #%1 importance = %2").arg(i + 1).arg(mat.at<float>(i, 0)));
}
}
return true;
}