git：https://github.com/linyi0604/Computer-Vision
1 # coding:utf-8
2
3 import cv2 4
5
6 # 检测i方框 包含o方框
7 def is_inside(o, i): 8 ox, oy, ow, oh = o 9 ix, iy, iw, ih = i 10 return ox > ix and ox + ow < ix + iw and oy + oh < iy + ih 11
12
13 # 将人外面的方框画出来
14 def draw_person(image, person): 15 x, y, w, h = person 16 cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 255), 2) 17
18
19 # 读入图片
20 img = cv2.imread("../data/people2.jpg") 21 # 获取hog检测器对象
22 hog = cv2.HOGDescriptor() 23 # 设置检测人的默认检测器
24 hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector()) 25 # 在图片中检测人，
26 # 返回found列表 每个元素是一个(x, y, w, h)的矩形，w是每一个矩形的置信度
27 found, w = hog.detectMultiScale(img) 28 found_filtered = [] 29 # 如果方框有包含，只留下内部的小方块
30 for ri, r in enumerate(found): 31 for qi, q in enumerate(found): 32 if ri != qi and is_inside(r, q): 33 break
34 else: 35 found_filtered.append(r) 36
37 # 将每一个方块画出来
38 for person in found_filtered: 39 draw_person(img, person) 40
41
42 cv2.imshow("person detection", img) 43 cv2.waitKey() 44 cv2.destroyAllWindows()
