-
Notifications
You must be signed in to change notification settings - Fork 0
/
objectdetection.py
43 lines (38 loc) · 1.34 KB
/
objectdetection.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
import imutils
import time
import cv2
vs=cv2.VideoCapture(0)
firstFrame=None
area=500
while True:
_,img=vs.read()
text='Normal'
img=imutils.resize(img,width=500)
grayImg=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
grayImg=cv2.GaussianBlur(grayimg,(21,21),0)
if firstFrame is None:
firstFrame=grayImg
continue
imgDiff=cv2.absdiff(firstFrame,grayImg)
threshImg = cv2.threshold(imgDiff, 25, 255, cv2.THRESH_BINARY)[1]
threshImg = cv2.dilate(threshImg,None,iterations=2)
cnts = cv2.findContours(threshImg.copy(), cv2.RETR_EXTERNAL,
cv2.CHAIN_APPROX_SIMPLE)
cnts=imutils.grab_contours(cnts)
for c in cnts:
if cv2.contourArea(c)<area:
continue
(x,y,w,h)=cv2.boundingRect(c)
cv2.rectangle(img,(x,y),(x+w,y+h),(0,255,0),2)
text="Moving Object detected"
print(text)
cv2.putText(img,text,(10,20),
cv2.FONT_HERSHEY_SIMPLEX,0.5,(0,0,255),2)
cv2.imshow("VideoStream",img)
cv2.imshow("Thresh",threshImg)
cv2.imshow("Image Difference",imgDiff)
key=cv2.waitKey(1)&0xFF
if key==ord("q"):
break
vs.release()
cv2.destroyAllWindows()