-
Notifications
You must be signed in to change notification settings - Fork 0
/
detect.py
26 lines (22 loc) · 1005 Bytes
/
detect.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import cv2
import imutils
# HOG feature descriptor is used in computer vision popularly for object detection
# detector
hog = cv2.HOGDescriptor()
hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())
cap = cv2.VideoCapture('pedestrain-crossing-video.mp4') #recorded video path copy
while cap.isOpened():
# Read the video stream
res, image = cap.read()
if res:
image = imutils.resize(image, width=min(400, image.shape[1]))
(regions, _) = hog.detectMultiScale(image, winStride=(4, 4), padding=(4, 4),scale=1.05)
for (x, y, w, h) in regions:
cv2.rectangle(image, (x, y),(x+w,y+h),(0,0,255), 2) #draw rectangle(coundary box) around the detected person
cv2.imshow("Image", image) #output video
if(cv2.waitKey(25) & 0xFF == ord('q')): #to stop the video press key 'q'
break
else:
break
cap.release()
cv2.destroyAllWindows()