python opencv3 检测人

Posted 2020-12-28 稀里糊涂林老冷

 

 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()