-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
66 lines (55 loc) · 1.85 KB
/
main.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
import glob
import os
import time
from threading import Thread
import cv2
def send_email(obj):
print('Email Sent')
def clean_folder():
images = glob.glob('images/*.png')
for image in images:
os.remove(image)
video = cv2.VideoCapture(0)
time.sleep(1)
first_frame = None
status_list = []
count = 1
while True:
status = 0
check, frame = video.read()
grey_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
grey_frame_gau = cv2.GaussianBlur(grey_frame, (21, 21), 0)
if first_frame is None:
first_frame = grey_frame_gau
delta_frame = cv2.absdiff(first_frame, grey_frame_gau)
thresh_frame = cv2.threshold(delta_frame, 60, 255,cv2.THRESH_BINARY)[1]
dil_frame = cv2.dilate(thresh_frame, None, iterations=2)
# cv2.imshow('My Video', dil_frame)
contours, check = cv2.findContours(dil_frame, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for contour in contours:
if cv2.contourArea(contour) < 10000:
continue
x, y, w, h = cv2.boundingRect(contour)
rectangle = cv2.rectangle(frame, (x, y), (w+x, y+h), (0, 255, 0), 3)
if rectangle.any():
status = 1
cv2.imwrite(f'images/{count}.png', frame)
count = count + 1
all_images = glob.glob('images/*.png')
index = int(len(all_images)/2)
image_with_object = all_images[index]
status_list.append(status)
status_list = status_list[-2:]
print(status_list)
if status_list[0] == 1 and status_list[1] == 0:
email_thread = Thread(target=send_email, args=(image_with_object, ))
email_thread.daemon = True
clean_thread = Thread(target=clean_folder)
clean_thread.daemon = True
email_thread.start()
cv2.imshow('Video', frame)
key = cv2.waitKey(1)
if key == ord('q'):
break
video.release()
clean_thread.start()