forked from Illinois-IoT/main-project
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdetectmotion.py
executable file
·77 lines (56 loc) · 1.85 KB
/
detectmotion.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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from imutils.video import VideoStream
import argparse
import datetime
import imutils
import time
import cv2
import RPi.GPIO as GPIO
GPIO.setmode(GPIO.BCM)
LED = 24
GPIO.setup(LED, GPIO.OUT)
GPIO.setwarnings(False)
ap = argparse.ArgumentParser()
ap.add_argument("-v", "--video", help="path to the video file")
ap.add_argument("-a", "--min-area", type = int, default = 500, help = "minimum area size")
args = vars(ap.parse_args())
if args.get("video", None) is None:
vs = VideoStream(src).start()
time.sleep(5.0)
else:
vs = cv2.VideoCapture(args["video"])
firstFrame = None
while True:
_, frame = vs.read()
text = "Unoccupied"
GPIO.output(LED,False)
if frame is None:
break
frame = imutils.resize(frame, width=500)
gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray,(21,21),0)
if firstFrame is None:
firstFrame = gray
continue
frameDelta = cv2.absdiff(firstFrame,gray)
thresh = cv2.threshold(frameDelta,25,255,cv2.THRESH_BINARY)[1]
thresh = cv2.dilate(thresh,None,iterations=2)
cnts = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL,
cv2.CHAIN_APPROX_SIMPLE)
cnts = imutils.grab_contours(cnts)
for c in cnts:
if cv2.contourArea(c) < args["min_area"]:
continue
(x,y,w,h) = cv2.boundingRect(c)
cv2.rectangle(frame,(x,y),(x+w,y+h), (0,255,0),2)
text = "Occupied"
GPIO.output(LED,True)
cv2.putText(frame, "Room Status: {}".format(text), (10,20), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,0,255),2)
cv2.putText(frame, datetime.datetime.now().strftime("%A %d %B %Y %I: %M: %S%p"), (10,frame.shape[0]-10), cv2.FONT_HERSHEY_SIMPLEX, 0.35,(0,0,255),1)
cv2.imshow("Security Feed", frame)
cv2.imshow("Thresh", thresh)
cv2.imshow("Frame Delta", frameDelta)
key = cv2.waitKey(1) & 0xFF
if key == ord("q"):
break
vs.stop() if args.get("video", None) is None else vs.release()
cv2.destroyAllWindows()