mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-08-06 14:36:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #4174 from yangj1e:fix_svm_trainauto

Browse Source
This commit is contained in:
Vadim Pisarevsky 2015-07-08 10:04:40 +00:00
commit 4997f5dfe9
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/ml/src/svm.cpp
View File

@ -1675,6 +1675,7 @@ public:
Mat samples = data->getTrainSamples(); Mat samples = data->getTrainSamples();
Mat responses; Mat responses;
bool is_classification = false; bool is_classification = false;
Mat class_labels0;
int class_count = (int)class_labels.total(); int class_count = (int)class_labels.total();
if( svmType == C_SVC || svmType == NU_SVC ) if( svmType == C_SVC || svmType == NU_SVC )
@ -1688,7 +1689,8 @@ public:
setRangeVector(temp_class_labels, class_count); setRangeVector(temp_class_labels, class_count);
// temporarily replace class labels with 0, 1, ..., NCLASSES-1 // temporarily replace class labels with 0, 1, ..., NCLASSES-1
Mat(temp_class_labels).copyTo(class_labels); class_labels0 = class_labels;
class_labels = Mat(temp_class_labels).clone();
} }
else else
responses = data->getTrainResponses(); responses = data->getTrainResponses();
@ -1821,6 +1823,7 @@ public:
} }
params = best_params; params = best_params;
class_labels = class_labels0;
return do_train( samples, responses ); return do_train( samples, responses );
} }