使用Python + OpenCV的HOG描述符

时间:2016-09-06 20:46:42

标签: python python-2.7 opencv image-processing opencv3.1

我正在尝试使用OpenCV实现HOG Descriptor来检测video中的行人。我目前正在使用OpenCV hogcascade_pedestrians.xml的预制数据集。虽然HOG Descriptor对人体检测非常有效,但是这部分的文档在互联网上非常差。我一直在用Python编写行人检测代码,我已停止使用以下代码:

import cv2
import numpy as np
import imutils

VidCap = cv2.VideoCapture('pedestrians.mp4')

HOGCascade = cv2.HOGDescriptor('hogcascade_pedestrians.xml')
HOGCascade.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())

while True:

    _ , image = VidCap.read()
    image = imutils.resize(image, width=700)

    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    clahe = cv2.createCLAHE(clipLimit=15.0,tileGridSize=(8,8))
    gray = clahe.apply(gray)

    winStride = (8,8)
    padding = (16,16)
    scale = 1.05
    meanshift = -1

    (rects, weights) = HOGCascade.detectMultiScale(gray, winStride=winStride,
                                            padding=padding,
                                            scale=scale,
                                            useMeanshiftGrouping=meanshift)

    for (x, y, w, h) in rects:
        cv2.rectangle(image, (x, y), (x+w, y+h), (0,200,255), 2)

    cv2.imshow('Image', image)

    if cv2.waitKey(5) == 27:
        break

VidCap.release()
cv2.destroyAllWindows()

我认为代码脚本将类似于为Haar Cascades编写的代码。但是我试过了,但是我遇到了错误。有没有人知道如何用Python在OpenCV上实现HOG描述符。

我已阅读以下question,但我从第二个答案中得不到任何结果。

我的问题是我无法找到编写代码的方法,因为有关此部分的文档很差。

注意:我在Python 2.7.11中使用OpenCV 3.1.0-dev

1 个答案:

答案 0 :(得分:0)

HOGCascade = cv2.HOGDescriptor()

如果你想使用这个.xml,你需要做很多准备工作。

当你最终得到可用的描述符时,你应该替换cv2.HOGDescriptor_getDefaultPeopleDetector() in setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())