http://blog.youkuaiyun.com/firefight/article/details/6400060
// svm.cpp : Defines the entry point for the console application.
//
#include "stdafx.h"
#include <cv.h>
#include <highgui.h>
using namespace std;
#include <ml.h>
void drawCross(cv::Mat &img, cv::Point center, cv::Scalar color)
{
int col = center.x > 2 ? center.x : 2;
int row = center.y> 2 ? center.y : 2;
line(img, cv::Point(col -2, row - 2), cv::Point(col + 2, row + 2), color);
line(img, cv::Point(col + 2, row - 2), cv::Point(col - 2, row + 2), color);
}
int newSvmTest(int rows, int cols, int testCount,std::string dst)
{
if(testCount > rows * cols)
return 0;
cv::Mat img = cv::Mat::zeros(rows, cols, CV_8UC3);
cv::Mat testPoint = cv::Mat::zeros(rows, cols, CV_8UC1);
cv::Mat data =cv::Mat::zeros(testCount, 2, CV_32FC1);
cv::Mat res = cv::Mat::zeros(testCount, 1, CV_32SC1);
//Create random test points
for (int i= 0; i< testCount; i++)
{
int row = rand() % rows;
int col = rand() % cols;
if(testPoint.at<unsigned char>(row, col) == 0)
{
testPoint.at<unsigned char>(row, col) = 1;
data.at<float>(i, 0) = float (col) / cols;
data.at<float>(i, 1) = float (row) / rows;
}
else
{
i--;
continue;
}
if (row > ( 50 * cos(col * CV_PI/ 100) + 200) )
{
drawCross(img, cv::Point(col, row), CV_RGB(255, 0, 0));
res.at<unsigned int>(i, 0) = 1;
}
else
{
if (col > 200)
{
drawCross(img, cv::Point(col, row), CV_RGB(0, 255, 0));
res.at<unsigned int>(i, 0) = 2;
}
else
{
drawCross(img, cv::Point(col, row), CV_RGB(0, 0, 255));
res.at<unsigned int>(i, 0) = 3;
}
}
}
imshow(dst, img);
//
// /////////////START SVM TRAINNING//////////////////
CvSVM svm = CvSVM();
CvSVMParams param;
CvTermCriteria criteria;
criteria= cvTermCriteria(CV_TERMCRIT_EPS, 1000, FLT_EPSILON);
param= CvSVMParams (CvSVM::C_SVC, CvSVM::RBF, 10.0, 8.0, 1.0, 10.0, 0.5, 0.1, NULL, criteria);
svm.train(data, res, cv::Mat(), cv::Mat(), param);
for (int i= 0; i< rows; i++)
{
for (int j= 0; j< cols; j++)
{
cv::Mat m = cv::Mat::zeros(1, 2, CV_32FC1);
m.at<float>(0,0) = float (j) / cols;
m.at<float>(0,1) = float (i) / rows;
float ret = 0.0;
ret = svm.predict(m);
cv::Scalar rcolor;
switch ((int) ret)
{
case 1: rcolor= CV_RGB(100, 0, 0); break;
case 2: rcolor= CV_RGB(0, 100, 0); break;
case 3: rcolor= CV_RGB(0, 0, 100); break;
}
line(img, cv::Point(j,i), cv::Point(j,i), rcolor);
}
}
cvShowImage("2",&IplImage(img));
// //Show support vectors
int sv_num= svm.get_support_vector_count();
for (int i= 0; i< sv_num; i++)
{
const float* support = svm.get_support_vector(i);
circle(img, cv::Point((int) (support[0] * cols), (int) (support[1] * rows)), 5, CV_RGB(200, 200, 200));
}
cvShowImage("3",&IplImage(img));
cv::waitKey(0);
return 0;
}
int main(int argc, char** argv)
{
cv::namedWindow("a",CV_WINDOW_NORMAL);
cv::moveWindow("a", 10, 600);
return newSvmTest(400, 600, 100,"a");
}