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

fixing issue #4969 of Itseez/opencv.

Browse Source 
Someone forgot to wrap the load function for SVMs in the
corresponding ml python module. Fixed that.
This commit is contained in:
Christoph Spörk 2016-01-04 15:47:26 +01:00
parent fade402899
commit a7aa198b4c
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/ml/include/opencv2/ml.hpp
View File

@ -719,6 +719,10 @@ public:
Use StatModel::train to train the model. Since %SVM has several parameters, you may want to Use StatModel::train to train the model. Since %SVM has several parameters, you may want to
find the best parameters for your problem, it can be done with SVM::trainAuto. */ find the best parameters for your problem, it can be done with SVM::trainAuto. */
CV_WRAP static Ptr<SVM> create(); CV_WRAP static Ptr<SVM> create();
CV_WRAP virtual void read( const FileNode& fn ) = 0;
CV_WRAP static Ptr<SVM> load(const String& fs);
}; };
/****************************************************************************************\ /****************************************************************************************\

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

@ -2261,6 +2261,17 @@ Ptr<SVM> SVM::create()
return makePtr<SVMImpl>(); return makePtr<SVMImpl>();
} }
Ptr<SVM> SVM::load(const String& filename)
{
FileStorage fs;
fs.open(filename, FileStorage::READ);
Ptr<SVM> svm = makePtr<SVMImpl>();
svm->read(fs.getFirstTopLevelNode());
return svm;
}
Mat SVM::getUncompressedSupportVectors() const Mat SVM::getUncompressedSupportVectors() const
{ {
const SVMImpl* this_ = dynamic_cast<const SVMImpl*>(this); const SVMImpl* this_ = dynamic_cast<const SVMImpl*>(this);