python opencv基于HOG-SVM行人检测

import numpy as np
import cv2
import matplotlib.pylab as plt

img = cv2.imread('./9781789343731_Code/images/66.jpg')
hog = cv2.HOGDescriptor()
hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())

(foundBoundingBoxes, weights) = \
    hog.detectMultiScale(img, hitThreshold=0, winStride=(4, 4),
                         padding=(2, 2), scale=1.02, finalThreshold=0)

    # hog.detectMultiScale(img, winStride=(4, 4), padding=(8,8),scale=1.02, finalThreshold=0.5)
print(len(foundBoundingBoxes))
imgWithRawBboxes = img.copy()

for (hx, hy, hw, hh) in foundBoundingBoxes:
    cv2.rectangle(imgWithRawBboxes, (hx, hy), (hx + hw, hy + hh), (255, 0, 255), 1)
plt.figure(figsize=(15, 12))
imgWithRawBboxes = cv2.cvtColor(imgWithRawBboxes, cv2.COLOR_BGR2RGB)
plt.imshow(imgWithRawBboxes, aspect='auto')
plt.axis('off')
plt.show()

from imutils.object_detection import non_max_suppression
rects = np.array([[x, y, x+w, y+h] for (x, y, w, h) in foundBoundingBoxes])
numsBoundingBoxes = non_max_suppression(rects, probs=None, overlapThresh=0.65)
print(len(rects), len(numsBoundingBoxes))

for (x1, y1, x2, y2) in numsBoundingBoxes:
    cv2.rectangle(img, (x1, y1), (x2, y2), (0, 0, 255), 2)
plt.figure(figsize=(15, 12))
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.imshow(img, aspect='auto')
plt.axis('off')
plt.show()