opencv利用svm训练
Posted lzq116
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了opencv利用svm训练相关的知识,希望对你有一定的参考价值。
#根据身高体重训练 预测男生还是女生
#1数据制作
rand1 = np.array([[155,48],[159,50],[164,53],[168,56],[172,60]])
rand2 = np.array([[152,53],[156,55],[160,56],[172,64],[176,65]])
label = np.array([[0],[0],[0],[0],[0],[1],[1],[1],[1],[1]])
data = np.vstack((rand1,rand2))
data = np.array(data,dtype = ‘float32‘)
print(data)
#2训练
svm = cv2.ml.SVM_create()#ml 机器学习模块
#属性设置
svm.setType(cv2.ml.SVM_C_SVC)
svm.setKernel(cv2.ml.SVM_LINEAR)
svm.setC(0.01)
#训练
result = svm.train(data,cv2.ml.ROW_SAMPLE,label)
#预测
pt_data = np.vstack([[167,55],[162,57]])
pt_data = np.array(pt_data,dtype = ‘float32‘)
print(pt_data)
(par1,par2) = svm.predict(pt_data)
print(par2)
以上是关于opencv利用svm训练的主要内容,如果未能解决你的问题,请参考以下文章
OpenCV-Python实战(番外篇)——利用 SVM 算法识别手写数字