SVM的C++实现

本文探讨使用C++实现支持向量机(SVM)以加深理解其内部工作原理。相较于Python中便捷的库调用,C++实现要求对SVM有更深入的了解。通过实践,有助于更好地掌握SVM。
摘要由CSDN通过智能技术生成

SVM看过很多次了,但是每一次都有收获。
用python实现SVM可以调用很多的包,所以对于SVM的内部实现很多都忽略了,但是用C++来实现SVM就要对SVM有足够的了解,如果希望更加理解SVM,我觉得用C++来实现一下SVM是一个很好的方法。
SVM的原理这里就不多解释了,很多博文都已经解释的很详细了。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include<string>
#include<fstream>
#include<sstream>

using std::sort;
using std::fabs;
using namespace std;

const int MAX_DIMENSION = 3;
const int MAX_SAMPLES = 306;
double x[MAX_SAMPLES][MAX_DIMENSION];
double y[MAX_SAMPLES];
double alpha[MAX_SAMPLES];
double w[MAX_DIMENSION];
double b;
double c;
double eps = 1e-6;


int num_samples = 306;
int num_dimension = 3;

struct _E {
        double val;
        int index;
}E[MAX_SAMPLES];

bool cmp(const _E & a, const _E & b)
{
        return a.val < b.val;
}



double max(double a, double b)
{
        return a > b ? a : b;
}

double min(double a, double b)
{
        return a > b ? b : a;
}

double kernal(double x1[], double x2[], double dimension)
{
        double ans = 0;
        for (int i = 0; i < dimension; i++)
        {
                ans += x1[i] * x2[i];
        }
        return ans;
}

double target_function()
{
        double ans = 0;
        for (int i = 0; i < num_samples; i++)
        {
                for (int j = 0; j < num_samples; j++)
                {
                        ans += alpha[i] * alpha[j] * y[i] * y[j] * kernal(x[i], x[j], num_dimension);
                }
        }

        for (int i = 0; i < num_samples; i++)
        {
                ans -= alpha[i];
        }

        return ans;
}


double g(double _x[], int dimension)
{
        double ans = b;

        for (int i = 0; i < num_samples; i++)
        {
                ans += alpha[i] * y[i] * kernal(x[i], _x, dimension);
        }

        return ans;
}

bool satisfy_constrains(int i, int dimension)
{
        if (alpha[i] == 0)
        {
                if (y[i] * g(x[i], dimension) >= 1)
                        return true;
                else
                        return false;
        }
        else if (alpha[i] > 0 && alpha[i] < c)
        {
                if (y[i] * g(x[i], dimension) == 1)
                        return true;
                else
                        return false;
        }
        else
        {
                if (y[i] * g(x[i], dimension) <= 1)
                        return true;
                else
                        return false;
        }
}


double calE(int i, int dimension)
{
        return g(x[i], dimension) - y[i];
}

void calW()
{
        for (int i = 0; i < num_dimension; i++)
        {
                w[i] = 0;
                for (int j = 0; j < num_samples; j++)
                {
                        w[i] += alpha[j] * y[j] * x[j][i];
                }
        }
        return;
}

void calB()
{
        double ans = y[0];
        for (int i = 0; i < num_samples; i++)
        {
                ans -= y[i] * alpha[i] * kernal(x[i], x[0], num_dimension);
        }
        b = ans;
        return;
}


void recalB(int alpha1index, int alpha2index, int dimension, double alpha1old, double alpha2old)
{
        double alpha1new = alpha[alpha1index];
        double alpha2new = alpha[alpha2index];

        al
HOG特征检测是一种基于梯度方向直方图的目标检测算法,可以用于人脸检测、行人检测等场景。下面是HOG特征检测+svmC++实现步骤: 1. 加载正负样本数据集并提取HOG特征 ```c++ #include <opencv2/opencv.hpp> #include <iostream> #include <fstream> #include <vector> using namespace std; using namespace cv; int main() { // 加载正负样本数据集 vector<Mat> pos_samples, neg_samples; string pos_path = "pos_samples/"; string neg_path = "neg_samples/"; for (int i = 0; i < 100; i++) { string img_name = pos_path + to_string(i) + ".jpg"; Mat img = imread(img_name, IMREAD_GRAYSCALE); pos_samples.push_back(img); } for (int i = 0; i < 100; i++) { string img_name = neg_path + to_string(i) + ".jpg"; Mat img = imread(img_name, IMREAD_GRAYSCALE); neg_samples.push_back(img); } // 提取HOG特征 HOGDescriptor hog(Size(64, 128), Size(16, 16), Size(8, 8), Size(8, 8), 9); vector<float> descriptors; vector<vector<float>> pos_descriptors, neg_descriptors; for (int i = 0; i < pos_samples.size(); i++) { hog.compute(pos_samples[i], descriptors); pos_descriptors.push_back(descriptors); } for (int i = 0; i < neg_samples.size(); i++) { hog.compute(neg_samples[i], descriptors); neg_descriptors.push_back(descriptors); } return 0; } ``` 2. 将HOG特征转换为svm的训练数据 ```c++ // 将HOG特征转换为svm的训练数据 Mat train_data; Mat train_label; for (int i = 0; i < pos_descriptors.size(); i++) { Mat row = Mat(pos_descriptors[i]).reshape(1, 1); train_data.push_back(row); train_label.push_back(1); } for (int i = 0; i < neg_descriptors.size(); i++) { Mat row = Mat(neg_descriptors[i]).reshape(1, 1); train_data.push_back(row); train_label.push_back(-1); } ``` 3. 训练svm模型并保存 ```c++ // 训练svm模型并保存 Ptr<ml::SVM> svm = ml::SVM::create(); svm->setType(ml::SVM::C_SVC); svm->setKernel(ml::SVM::LINEAR); svm->setTermCriteria(TermCriteria(TermCriteria::MAX_ITER, 100, 1e-6)); svm->train(train_data, ml::ROW_SAMPLE, train_label); svm->save("svm_model.xml"); ``` 4. 加载svm模型并进行目标检测 ```c++ // 加载svm模型并进行目标检测 Ptr<ml::SVM> svm = ml::StatModel::load<ml::SVM>("svm_model.xml"); HOGDescriptor hog(Size(64, 128), Size(16, 16), Size(8, 8), Size(8, 8), 9); Mat img = imread("test.jpg", IMREAD_GRAYSCALE); vector<Rect> found; vector<float> descriptors; hog.compute(img, descriptors); Mat row = Mat(descriptors).reshape(1, 1); float result = svm->predict(row); if (result == 1) { found.push_back(Rect(0, 0, 64, 128)); } hog.detectMultiScale(img, found, 0, Size(8, 8), Size(0, 0), 1.05, 2); for (int i = 0; i < found.size(); i++) { rectangle(img, found[i], Scalar(0, 0, 255), 2); } imshow("result", img); waitKey(0); ``` 以上就是HOG特征检测+svmC++实现步骤。需要注意的是,opencv中的svm模型只支持二分类,因此在本例中将正样本标签设为1,负样本标签设为-1。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值