用OPEN CV SVM 的例子程序

来源：互联网发布：mpv播放器 mac下载编辑：程序博客网时间：2024/05/23 13:53

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <opencv2/ml/ml.hpp>
#include <iostream>
using namespace std;

using namespace cv;

int main()
{
int width=512,height=512;
Mat image=Mat::zeros(height,width,CV_8UC3);

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 para
CvSVMParams params;
params.svm_type=CvSVM::C_SVC;
params.kernel_type=CvSVM::LINEAR;
params.term_crit=cvTermCriteria(CV_TERMCRIT_ITER,100,1E-6);

//TRAIM THE svm
CvSVM SVM;
SVM.train(trainingDatamat,labelsMat,Mat(),Mat(),params);

Vec3b green(0,255,0),blue(255,0,0);
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;

}

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);
thickness=2;
lineType=8;

int c=SVM.get_support_vector_count();
cout<<c;
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);
imshow("Sample Image",image);
waitKey(0);
}

0 0