[英]Tutorials for SVM with c++
我正在嘗試使用SVM使用c ++進行對象檢測。 我正在關注這個答案 。 我面臨的一個問題是,到目前為止,尚未使用CvSVM。 因此,我對培訓代碼進行了如下修改。
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/ml/ml.hpp>
using namespace cv;
using namespace cv::ml;
int main()
{
// Data for visual representation
int width = 512, height = 512;
Mat image = Mat::zeros(height, width, CV_8UC3);
// Set up training data
float labels[4] = {1.0, -1.0, -1.0, -1.0};
Mat labelsMat(4, 1, CV_32FC1, labels);
float trainingData[4][2] = { {501, 10}, {255, 10}, {501, 255}, {10, 501} };
Mat trainingDataMat(4, 2, CV_32FC1, trainingData);
// Set up SVM's parameters
Ptr<SVM> svm = SVM::create();
svm->setType(SVM::C_SVC);
svm->setKernel(SVM::LINEAR);
//svm.term_crit = SVM::getTermCriteria(CV_TERMCRIT_ITER, 100, 1e-6);
// Train the SVM
//Ptr<SVM> svm1 = SVM::trainAuto();
SVM->train(trainingDataMat, labelsMat, Mat(), Mat(), svm);
Vec3b green(0,255,0), blue (255,0,0);
// Show the decision regions given by the SVM
for (int i = 0; i < image.rows; ++i)
for (int j = 0; j < image.cols; ++j)
{
Mat sampleMat = (Mat_<float>(1,2) << j,i);
float response = svm->predict(sampleMat);
if (response == 1)
image.at<Vec3b>(i,j) = green;
else if (response == -1)
image.at<Vec3b>(i,j) = blue;
}
// Show the training data
int thickness = -1;
int lineType = 8;
circle( image, Point(501, 10), 5, Scalar( 0, 0, 0), thickness, lineType);
circle( image, Point(255, 10), 5, Scalar(255, 255, 255), thickness, lineType);
circle( image, Point(501, 255), 5, Scalar(255, 255, 255), thickness, lineType);
circle( image, Point( 10, 501), 5, Scalar(255, 255, 255), thickness, lineType);
// Show support vectors
thickness = 2;
lineType = 8;
int c = SVM.get_support_vector_count();
for (int i = 0; i < c; ++i)
{
const float* v = SVM.get_support_vector(i);
circle( image, Point( (int) v[0], (int) v[1]), 6, Scalar(128, 128, 128), thickness, lineType);
}
imwrite("result.png", image); // save the image
imshow("SVM Simple Example", image); // show it to the user
waitKey(0);
}
我無法實現火車功能。 它說找不到該功能。 請幫助我使用此代碼的較新版本。
您的代碼中有幾個錯誤。 有些是C ++語法錯誤,有些是由於您使用的是OpenCV 2.4.X api而與OpenCV 3.0不同
1)在引用svm實例時,應使用svm
(變量名),而不是SVM
的類名。
2)在分類問題的情況下,回答必須是分類的。 所以
float labels[4] = { 1.0, -1.0, -1.0, -1.0 };
Mat labelsMat(4, 1, CV_32FC1, labels);
應該:
int labels[4] = { 1, -1, -1, -1 };
Mat labelsMat(4, 1, CV_32SC1, labels);
3) train
接受不同的參數。 SVM->train(trainingDataMat, labelsMat, Mat(), Mat(), svm);
應為: svm->train(trainingDataMat, ROW_SAMPLE, labelsMat);
4) get_support_vector_count
在OpenCV 3.0中不存在。 int c = SVM.get_support_vector_count();
應該是: int c = svm->getSupportVectors().rows;
5) get_support_vector
在OpenCV 3.0中不存在。 const float* v = SVM.get_support_vector(i);
應該是: const float* v = svm->getSupportVectors().ptr<float>(i);
此答案中的代碼已按預期工作。 如果您引入此類錯誤,則顯然將無法正常工作。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.