generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed CvRTrees::read in case nactive_vars==var_count (thanks to Malcolm Reynolds)

Browse Source
This commit is contained in:
Maria Dimashova
2011-11-10 13:40:11 +00:00
parent 61515f78c3
commit a3ce652eb4
1 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -813,11 +813,16 @@ void CvRTrees::read( CvFileStorage* fs, CvFileNode* fnode )
active_var_mask = cvCreateMat( 1, var_count, CV_8UC1 ); active_var_mask = cvCreateMat( 1, var_count, CV_8UC1 );
{ {
// initialize active variables mask // initialize active variables mask
CvMat submask1, submask2; CvMat submask1;
cvGetCols( active_var_mask, &submask1, 0, nactive_vars ); cvGetCols( active_var_mask, &submask1, 0, nactive_vars );
cvGetCols( active_var_mask, &submask2, nactive_vars, var_count );
cvSet( &submask1, cvScalar(1) ); cvSet( &submask1, cvScalar(1) );
cvZero( &submask2 );
if( nactive_vars < var_count )
{
CvMat submask2;
cvGetCols( active_var_mask, &submask2, nactive_vars, var_count );
cvZero( &submask2 );
}
} }
} }