Skip to content

Commit bdd093b

Browse files
authored
Merge pull request #3782 from ivafanas:speedup_random_forest_getvotes
speedup random forest getVotes method.
2 parents 328b1b1 + 2579b1e commit bdd093b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

modules/ml/src/rtrees.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -388,9 +388,10 @@ class DTreesImplForRTrees CV_FINAL : public DTreesImpl
388388
results = output.getMat();
389389
for( i = 0; i < nsamples; i++ )
390390
{
391+
const Mat sampleRow = samples.row(i);
391392
for( j = 0; j < ntrees; j++ )
392393
{
393-
float val = predictTrees( Range(j, j+1), samples.row(i), flags);
394+
float val = predictTrees( Range(j, j+1), sampleRow, flags);
394395
results.at<float> (i, j) = val;
395396
}
396397
}
@@ -408,9 +409,10 @@ class DTreesImplForRTrees CV_FINAL : public DTreesImpl
408409
for( i = 0; i < nsamples; i++ )
409410
{
410411
votes.clear();
412+
const Mat sampleRow = samples.row(i);
411413
for( j = 0; j < ntrees; j++ )
412414
{
413-
int val = (int)predictTrees( Range(j, j+1), samples.row(i), flags);
415+
int val = (int)predictTrees( Range(j, j+1), sampleRow, flags);
414416
votes.push_back(val);
415417
}
416418

0 commit comments

Comments
 (0)