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

Add img/vid to gitignore; minor changes in video/image presence

Browse Source
This commit is contained in:
Lennart Heimbs 2019-08-04 09:14:46 +02:00
parent 93c203c86f
commit cc9304365a
3 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -4,3 +4,6 @@ venv/
camera/venv
camera/images
camera/videos
*.jpg
*.h264
*.mp4

5
camera/image_presence.py
View File

@ -49,6 +49,7 @@ for imagePath in paths.list_images(args["images"]):
print("[INFO] {}: {} original boxes, {} after suppression".format(filename, len(rects), len(pick)))
# show the output images
if len(pick):
#cv2.imshow("Before NMS", orig)
#cv2.imshow("After NMS", image)
#cv2.waitKey(0)
cv2.imshow("After NMS", image)
cv2.waitKey(0)

5
camera/video_presence.py
View File

@ -50,6 +50,7 @@ if args.get("video", None) is None:
# otherwise, we are reading from a video file
else:
vs = cv2.VideoCapture(args["video"])
#vs.set(cv2.CAP_PROP_FPS, 2)
"""" Analyzing video frames """
# loop over the frames of the video, and store corresponding information from each frame
@ -92,7 +93,7 @@ while True:
# dilate the thresholded image to fill in holes, then find contours on thresholded image
thresh = cv2.dilate(thresh, None, iterations=2)
thresh = np.uint8(thresh)
cnts, im2 = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
_, cnts, im2 = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
#cnts = cnts if imutils.is_cv2() else im2
#print(len(cnts))
#if len(cnts) > 1:
@ -120,7 +121,7 @@ while True:
trackbox = frame[y:y+h, x:x+w]
trackbox = cv2.resize(trackbox, (224, 224))
cv2.imshow('image',trackbox)
#cv2.imshow('image',trackbox)
blob = cv2.dnn.blobFromImage(cv2.resize(trackbox, (300, 300)),0.007843, (300, 300), 127.5)
net.setInput(blob)
detections = net.forward()