mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-27 12:40:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

speedup random forest getVotes method.

Browse Source
This commit is contained in:
Ivan Afanasyev 2024-08-20 15:35:59 +07:00
parent ae4a11b0c0
commit 7d3d2101f9
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/ml/src/rtrees.cpp
View File

@ -388,9 +388,10 @@ public:
results = output.getMat();
for( i = 0; i < nsamples; i++ )
{
const Mat sampleRow = samples.row(i);
for( j = 0; j < ntrees; j++ )
{
float val = predictTrees( Range(j, j+1), samples.row(i), flags);
float val = predictTrees( Range(j, j+1), sampleRow, flags);
results.at<float> (i, j) = val;
}
}
@ -408,9 +409,10 @@ public:
for( i = 0; i < nsamples; i++ )
{
votes.clear();
const Mat sampleRow = samples.row(i);
for( j = 0; j < ntrees; j++ )
{
int val = (int)predictTrees( Range(j, j+1), samples.row(i), flags);
int val = (int)predictTrees( Range(j, j+1), sampleRow, flags);
votes.push_back(val);
}