fixed read_train_data_params(): always read cat_count & cat_map in case of classifica...

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

diff --git a/opencv/src/ml/mltree.cpp b/opencv/src/ml/mltree.cpp
index 91b6a70d4eae51be384e5a53eb4743beb7c9c9e1..1cecfbb5b136928c1deda0183f819d372e349912 100644 (file)
--- a/opencv/src/ml/mltree.cpp
+++ b/opencv/src/ml/mltree.cpp
@@ -3055,7 +3055,7 @@ void CvDTree::read_train_data_params( CvFileStorage* fs, CvFileNode* node )
         CV_ERROR( CV_StsParseError, "var_type is inconsistent with cat_var_count and ord_var_count" );
     //////
 
-    if( data->cat_var_count > 0 )
+    if( data->cat_var_count > 0 || is_classifier )
     {
         int ccount, max_c_count = 0, total_c_count = 0;
         CV_CALL( data->cat_count = (CvMat*)cvReadByName( fs, node, "cat_count" ));