opencv学习历程002(opencv3中支持向量机的用法)

2017-09-21 11:19:47来源:CSDN作者:playezio人点击

分享

这些天一直在学习《opencv3编程入门》这本书,由于里面是opencv2的版本与32位的环境,总是得考出来新建一个空项目来学习例程,有一些收获。


今天学习到了第二章中支持向量机的部分,由于opencv3以及没有了CvSVM的用法,所以程序总是出现“未定义标识符cvsvm”这样的错误,于是我从博客http://www.cnblogs.com/denny402/p/5019233.html

中找到了最新的用法,代码如下:


#include "opencv2/opencv.hpp"
using namespace cv;
using namespace cv::ml;


int main(int, char**)
{
    int width = 512, height = 512;
    Mat image = Mat::zeros(height, width, CV_8UC3);  //创建窗口可视化


    // 设置训练数据
    int labels[10] = { 1, -1, 1, 1,-1,1,-1,1,-1,-1 };
    Mat labelsMat(10, 1, CV_32SC1, labels);


    float trainingData[10][2] = { { 501, 150 }, { 255, 10 }, { 501, 255 }, { 10, 501 }, { 25, 80 },
    { 150, 300 }, { 77, 200 } , { 300, 300 } , { 45, 250 } , { 200, 200 } };
    Mat trainingDataMat(10, 2, CV_32FC1, trainingData);


    // 创建分类器并设置参数
    Ptr<SVM> model =SVM::create();
    model->setType(SVM::C_SVC);  
    model->setKernel(SVM::LINEAR);  //核函数


    //设置训练数据 
    Ptr<TrainData> tData =TrainData::create(trainingDataMat, ROW_SAMPLE, labelsMat);


    // 训练分类器
    model->train(tData);


    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 = model->predict(sampleMat);  //进行预测,返回1或-1


        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;
    Scalar c1 = Scalar::all(0); //标记为1的显示成黑点
    Scalar c2 = Scalar::all(255); //标记成-1的显示成白点
    //绘图时,先宽后高,对应先列后行
    for (int i = 0; i < labelsMat.rows; i++)
    {
        const float* v = trainingDataMat.ptr<float>(i); //取出每行的头指针
        Point pt = Point((int)v[0], (int)v[1]);
        if (labels[i] == 1)
            circle(image, pt, 5, c1, thickness, lineType); 
        else
            circle(image, pt, 5, c2, thickness, lineType);
        
    }


    imshow("SVM Simple Example", image); 
    waitKey(0);


}

原文中第一行是“#include "stdafx.h"”,去掉后才可成功运行,运行后的图片如下:


接下来要开始编译opencv源代码的学习啦,加油!


微信扫一扫

第七城市微信公众平台