使用OpenCV进行SVM分类demo

代码来源 https://github.com/mbeyeler/opencv-machine-learning/blob/master/notebooks/06.01-Implementing-Your-First-Support-Vector-Machine.ipynb

from sklearn import datasets
import cv2
import matplotlib.pyplot as plt
import numpy as np
from sklearn import model_selection as ms
from sklearn import metricsX, y = datasets.make_classification(n_samples=100, n_features=2,n_redundant=0, n_classes=2,random_state=7816)plt.style.use('ggplot')
plt.set_cmap('jet')plt.figure(figsize=(10, 6))
plt.scatter(X[:, 0], X[:, 1], c=y, s=100)
plt.xlabel('x values')
plt.ylabel('y values')X = X.astype(np.float32)
y = y * 2 - 1X_train, X_test, y_train, y_test = ms.train_test_split(X, y, test_size=0.2, random_state=42
)def plot_decision_boundary(svm, X_test, y_test):# create a mesh to plot inh = 0.02  # step size in meshx_min, x_max = X_test[:, 0].min() - 1, X_test[:, 0].max() + 1y_min, y_max = X_test[:, 1].min() - 1, X_test[:, 1].max() + 1xx, yy = np.meshgrid(np.arange(x_min, x_max, h),np.arange(y_min, y_max, h))X_hypo = np.c_[xx.ravel().astype(np.float32),yy.ravel().astype(np.float32)]_, zz = svm.predict(X_hypo)zz = zz.reshape(xx.shape)plt.contourf(xx, yy, zz, cmap=plt.cm.coolwarm, alpha=0.8)plt.scatter(X_test[:, 0], X_test[:, 1], c=y_test, s=200)kernels = [cv2.ml.SVM_LINEAR, cv2.ml.SVM_INTER, cv2.ml.SVM_SIGMOID, cv2.ml.SVM_RBF]plt.figure(figsize=(14, 8))
for idx, kernel in enumerate(kernels):svm = cv2.ml.SVM_create()svm.setKernel(kernel)svm.train(X_train, cv2.ml.ROW_SAMPLE, y_train)_, y_pred = svm.predict(X_test)plt.subplot(2, 2, idx + 1)plot_decision_boundary(svm, X_test, y_test)plt.title('accuracy = %.2f' % metrics.accuracy_score(y_test, y_pred))
plt.show()

结果如下：

使用OpenCV进行SVM分类demo相关推荐

opencv C++ SVM模型训练与分类实现
最近想学习一下分类算法的内容,恰好opencv有SVM的函数,故先从这个下手.找了许多资料,发现要么是opencv2.3的,要么就没有具体实现代码,学习还是把代码与原理一起结合来看比较好. 其中,我主 ...
C3D：视频动作分类demo实现
C3D:视频动作分类demo实现 C3D这个网络是来自于learning Spatiotemporal feature with 3DConvolutional Networks这篇文章,我也对这篇文 ...
用OpenCV的SVM实现简单的手势识别（切水果）[附源码]
基于HOG特征的SVM分类器实现在机器学习领域,支持向量机SVM(Support Vector Machine)是一个有监督的学习模型,通常用来进行模式识别.分类.以及回归分析. 方向梯度直方图(H ...
【全解】基于OpenCv的SVM实现车牌检测与识别
都说深度学习的出现极力地打压着传统机器学习算法的地位,作为一个二刷机器学习经典算法的小伙伴告诉你:还真多半是这样,咳,毕竟还是差距较大,深度学习处理真实世界多维度的问题更权威!不过,有的事情还是机器学 ...
OpenCV支持向量机SVM简介
OpenCV支持向量机SVM简介支持向量机SVM简介目标什么是SVM? 最佳超平面如何计算? 源代码解释设置训练数据设置SVM的参数由SVM分类的区域支持向量结果支持向量机SVM简 ...
机器学习分类器——案例（opencv sklearn svm ann）
机器学习分类器--案例(opencv sklearn svm ann python) ps:最近师姐给我们留了一个任务,记录一下从一开始的什么都不懂到现在把任务做出来,并从中学习到的东西吧.... 语 ...
基于OpenCV的 SVM算法实现数字识别(四)---代码实现
三.基于SVM算法实现手写数字识别作为一个工科生,而非数学专业的学生,我们研究一个算法,是要将它用于实际领域的.下面给出基于OpenCV3.0的SVM算法手写数字识别程序源码(参考http://bl ...
LBP特征的实现及LBP+SVM分类
前言 LBP这篇博客发表了有一年多的时间了,当时是为了研究生毕业论文实验而写的,后来稍微总结了一下写了这篇博客,一年多时间里,大家提了一些宝贵的修改意见,这两天将代码重构了一下,结构更加简洁清晰,速度 ...
opencv c++ svm分类器初学
c++ 初学者 + opencv初学者写了一个简单的svm分类器,由于有很多不是很了解,所以代码注释得比较详细 svm 直接上代码吧大家将就看 #include "pch.h" ...

使用OpenCV进行SVM分类demo

使用OpenCV进行SVM分类demo相关推荐

最新文章

热门文章