fixed traincascade

This commit is contained in:
Maria Dimashova 2010-12-17 14:24:59 +00:00
parent 7767038ef0
commit e5c5a1cb3d

View File

@ -457,10 +457,18 @@ const int* CvCascadeBoostTrainData::get_cat_var_data( CvDTreeNode* n, int vi, in
catValuesBuf[i] = (int) valCache.at<float>( vi, sampleIndices[i]); catValuesBuf[i] = (int) valCache.at<float>( vi, sampleIndices[i]);
} }
else else
{
if( vi >= numPrecalcVal && vi < var_count )
{ {
for( int i = 0; i < nodeSampleCount; i++ ) for( int i = 0; i < nodeSampleCount; i++ )
catValuesBuf[i] = (int)(*featureEvaluator)( vi, sampleIndices[i] ); catValuesBuf[i] = (int)(*featureEvaluator)( vi, sampleIndices[i] );
} }
else
{
get_cv_labels( n, catValuesBuf );
}
}
return catValuesBuf; return catValuesBuf;
} }