heimbsle69869/smarthome-presence-detect
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add Count people

Browse Source
This commit is contained in:
lheimbs 2019-08-01 10:24:44 +02:00
parent a89e27abfc
commit fb7560d92e
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
camera/video_presence.py
View File

@ -60,8 +60,10 @@ differ = None
now = '' now = ''
framecounter = 0 framecounter = 0
trackeron = 0 trackeron = 0
people_count_total = 0
while True: while True:
people_count_per_frame = 0
frame = vs.read() frame = vs.read()
frame = frame if args.get("video", None) is None else frame[1] frame = frame if args.get("video", None) is None else frame[1]
# if the frame can not be grabbed, then we have reached the end of the video # if the frame can not be grabbed, then we have reached the end of the video
@ -126,9 +128,11 @@ while True:
for i in np.arange(0, detections.shape[2]): for i in np.arange(0, detections.shape[2]):
confidence = detections[0, 0, i, 2] confidence = detections[0, 0, i, 2]
confidence_level = 0.7 confidence_level = 0.8
if confidence > confidence_level: if confidence > confidence_level:
people_count_per_frame+=1
people_count_total+=1
# extract the index of the class label from the `detections`, then compute the (x, y)-coordinates of # extract the index of the class label from the `detections`, then compute the (x, y)-coordinates of
# the bounding box for the object # the bounding box for the object
idx = int(detections[0, 0, i, 1]) idx = int(detections[0, 0, i, 1])
@ -156,7 +160,7 @@ while True:
# check to see if we are currently tracking an object, if so, ignore other boxes # check to see if we are currently tracking an object, if so, ignore other boxes
# this code is relevant if we want to identify particular persons (section 2 of this tutorial) # this code is relevant if we want to identify particular persons
if initBB2 is not None: if initBB2 is not None:
# grab the new bounding box coordinates of the object # grab the new bounding box coordinates of the object
@ -183,6 +187,8 @@ while True:
info = [ info = [
("Success", "Yes" if success else "No"), ("Success", "Yes" if success else "No"),
("FPS", "{:.2f}".format(fps.fps())), ("FPS", "{:.2f}".format(fps.fps())),
("People Frame", "{}".format(people_count_per_frame)),
("People Total", "{}".format(people_count_total))
] ]
# loop over the info tuples and draw them on our frame # loop over the info tuples and draw them on our frame