图像处理（五）

1、SVM

#SVM：寻求一个最优超平面
#SVM核：line
#身高、体重、性别
import cv2
import numpy as np
import matplotlib.pyplot as plt

#数据集
data1 = np.array([[155,48],[159,50],[160,51],[165,54],[170,58]])
data2 = np.array([[155,52],[159,56],[160,57],[165,60],[170,63]])
#label
label = np.array([[0],[0],[0],[0],[0],[1],[1],[1],[1],[1]])
#合并数据集
#np.vstack(data1,data2):在竖直方向上堆叠 np.hstack在水平方向铺开   data1必须shi
data = np.vstack((data1,data2))
data = np.array(data,np.float32)
print(data)
#训练准备
svm = cv2.ml.SVM_create()
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([[155,45],[156,57]])
pt_data = np.array(pt_data,np.float32)
print(pt_data)
(par1,par2) = svm.predict(pt_data)
print(par2)