svm python opencv_Python-OpenCV-SVM实现

要使用OpenCV机器学习算法，必须编写一些包装类：

1。第一父类class StatModel(object):

'''parent class - starting point to add abstraction'''

def load(self, fn):

self.model.load(fn)

def save(self, fn):

self.model.save(fn)

2。最后是SvM包装器：class SVM(StatModel):

'''wrapper for OpenCV SimpleVectorMachine algorithm'''

def __init__(self):

self.model = cv2.SVM()

def train(self, samples, responses):

#setting algorithm parameters

params = dict( kernel_type = cv2.SVM_LINEAR,

svm_type = cv2.SVM_C_SVC,

C = 1 )

self.model.train(samples, responses, params = params)

def predict(self, samples):

return np.float32( [self.model.predict(s) for s in samples])

3.示例用法：import numpy as np

import cv2

samples = np.array(np