mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 22:44:02 +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 responses;
bool is_classification = false;
Mat class_labels0;
int class_count = (int)class_labels.total();
if( svmType == C_SVC || svmType == NU_SVC )
@ -1688,7 +1689,8 @@ public:
setRangeVector(temp_class_labels, class_count);
// 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
responses = data->getTrainResponses();
@ -1821,6 +1823,7 @@ public:
}
params = best_params;
class_labels = class_labels0;
return do_train( samples, responses );
}