Opencv 3 SVM training

后端 未结 3 1889
一整个雨季
一整个雨季 2020-12-13 16:34

As you may know, many things changed in OpenCV 3 (in comparision to the openCV2 or the old first version).

In the old days, to train SVM one would use:



        
3条回答
  •  情深已故
    2020-12-13 16:42

    I was porting my code from OpenCV 2.4.9 to 3.0.0-rc1 and had the same issue. Unfortunately the API has changes since the answer was posted, so I would like to update it accordingly:

    Ptr svm = ml::SVM::create();
    svm->setType(ml::SVM::C_SVC);
    svm->setKernel(ml::SVM::POLY);
    svm->setGamma(3);
    
    Mat trainData; // one row per feature
    Mat labels;    
    Ptr tData = ml::TrainData::create(trainData, ml::SampleTypes::ROW_SAMPLE, labels);
    svm->train(tData);
    // ...
    Mat query; // input, 1channel, 1 row (apply reshape(1,1) if nessecary)
    Mat res;   // output
    svm->predict(query, res);
    

提交回复
热议问题