generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix SVM::trainAuto (Bug #4464)

Browse Source 
Fix SVM::trainAuto (Bug #4464)
This commit is contained in:
Jie Yang 2015-07-05 22:52:10 +09:30
parent f4cb8c0dfb
commit 2ba0d31785
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 );
}