fixed dtree

git-svn-id: https://code.ros.org/svn/opencv/trunk@2944 73c94f0f-984f-4a5f-82bc-2d8db8d8ee08

diff --git a/opencv/src/ml/mltree.cpp b/opencv/src/ml/mltree.cpp
index 47d61a334ff7947849495060b49190640d9d95af..858685bbd11d2bde998820d7b9303fc73e30daa0 100644 (file)
--- a/opencv/src/ml/mltree.cpp
+++ b/opencv/src/ml/mltree.cpp
@@ -1701,7 +1701,7 @@ double CvDTree::calc_node_dir( CvDTreeNode* node )
 
     if( data->get_var_type(vi) >= 0 ) // split on categorical var
     {
-        cv::AutoBuffer<int> inn_buf(n*(data->have_priors ? 1 : 2));
+        cv::AutoBuffer<int> inn_buf(n*(!data->have_priors ? 1 : 2));
         int* labels_buf = (int*)inn_buf;
         const int* labels = data->get_cat_var_data( node, vi, labels_buf );
         const int* subset = node->split->subset;