SVM + HOG特征实现图像物体检测

'''
SVM + HOG 实现小狮子识别分类
'''
import cv2
import  numpy as np
import  matplotlib.pyplot as plt
'''
定义参数
'''
#样本图片大小 正负样本个数 正:负 = 1:3左右 block大小 block步长 cell大小 bin的个数
winSize = (64,128)
posNum = 820
negNum = 1931
blockSize = (16,16)
blockStride = (8,8)
cellSize = (8,8)
nBin = 9
#创建hog特征 svm
hog = cv2.HOGDescriptor(winSize,blockSize,blockStride,cellSize,nBin)
svm = cv2.ml.SVM_create()
#计算hog特征
#特征数量
featureNum = int(((128-16)/8+1)*((64-16)/8+1)*4*9)  #3780
featureArray = np.zeros(((posNum+negNum),featureNum),np.float32)
labelArray = np.zeros(((posNum+negNum),1),np.int32)
#svm 监督学习 学习的是每个image的hog特征
#标注正样本标签 标成1
for i in range(0,posNum):
    fileName = 'G:\\images\\pos\\'+str(i+1)+'.jpg'
    img = cv2.imread(fileName)
    #计算每张图片的hog特征
    hist = hog.compute(img,(8,8)) #3780
    for j in range(0,featureNum):
        featureArray[i,j] = hist[j]
    labelArray[i,0] = 1
#标注负样本标签 标成-1
for i in range(0,negNum):
    fileName = 'G:\\images\\neg\\' + str(i + 1) + '.jpg'
    img = cv2.imread(fileName)
    # 计算每张图片的hog特征
    hist = hog.compute(img, (8, 8))  # 3780
    for j in range(0,featureNum):
        featureArray[i+posNum,j] = hist[j]
    labelArray[i+posNum,0] = -1
#设置svm的参数
svm.setType(cv2.ml.SVM_C_SVC)
svm.setKernel(cv2.ml.SVM_LINEAR)
svm.setC(0.01)
#svm训练
ret = svm.train(featureArray,cv2.ml.ROW_SAMPLE,labelArray)
#检测
alpha = np.zeros((1),np.float32)
rho = svm.getDecisionFunction(0,alpha)
alphaArray = np.zeros((1,1),np.float32)
supportVArray = np.zeros((1,featureNum),np.float32)
resultArray = np.zeros((1,featureNum),np.float32)
alphaArray[0,0] = alpha
#detect
myDetect = np.zeros((3781),np.float32)
for i in range(0,3780):
    myDetect[i] = resultArray[0,i]
myDetect[3780] = rho[0]
#构建hog
myHog = cv2.HOGDescriptor()
myHog.setSVMDetector(myDetect)
imageSrc = cv2.imread("G:\\images\\Test2.jpg",1)
objs = myHog.detectMultiScale(imageSrc,0,(8,8),(32,32),1.05,2)
x = int(objs[0][0][0])
y = int(objs[0][0][1])
w = int(objs[0][0][2])
h = int(objs[0][0][3])
cv2.rectangle(imageSrc,(x,y),(x+w,y+h),(255,0,0),2)
cv2.imshow('dst',imageSrc)
cv2.waitKey(0)