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

pull/13383/head
Maria Dimashova 14 years ago
parent 61515f78c3
commit a3ce652eb4
  1. 13
      modules/ml/src/rtrees.cpp

@ -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 );
}
} }
} }

Loading…
Cancel
Save