opencv4下使用SVM进行简单颜色分类

brdlizg

于 2019-11-28 11:44:07 发布

阅读量2.1k

点赞数 2

分类专栏： opencv 文章标签：颜色识别

本文链接：https://blog.csdn.net/brdlizg/article/details/103290448

版权

本文介绍了在OpenCV4环境下利用SVM进行颜色分类的工作。作者参照了相关资料，实现了从彩色图片中提取红色的能力，但遇到问题：测试样本变更会导致训练失败，可能与所选内核类型有关。尽管能识别红色，但也存在误判，如将黄色和部分绿色误判为红色。

摘要由CSDN通过智能技术生成

最近需要进行颜色分类的工作，查阅资料并参考 https://blog.csdn.net/ZaiYxx/article/details/50614006，修改成可在opencv4下运行的代码。

#include <opencv2/core.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/ml.hpp>
#include <iostream>

int svmColorRed()
{
#if 1
//     initial SVM
    cv::Ptr<cv::ml::SVM> svm = cv::ml::SVM::create();
    // 训练
    {
        int labels[6] = {1, 1, 1, 0, 0, 0};
        float trainingData[6][3] ={ {255,0, 0}, {255,0,0}, {255, 0,0}, {0,0,255 },{0,0,255},{0,0,255}};
        cv::Mat train_DataMat(6, 3, CV_32FC1, trainingData);
        cv::Mat labelsMat(6, 1, CV_32SC1, labels);
        
        svm->setType(cv::ml::SVM::Types::C_SVC);
        svm->setKernel(cv::ml::SVM::KernelTypes::LINEAR);
        svm->setTermCriteria(cv::TermCriteria(cv::TermCriteria::MAX_ITER, 100, 1e-6))