-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
71 lines (55 loc) · 1.89 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
67
68
69
70
71
import cv2
import time
import glob
import os
from threading import Thread
from emailing import send_email
video = cv2.VideoCapture(1)
time.sleep(1)
first_frame = None
status_list = []
count = 1
def clean_folder():
time.sleep(8)
images = glob.glob("images/*.png")
for image in images:
os.remove(image)
while True:
status = 0
check, frame = video.read()
gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
gray_frame_gau = cv2.GaussianBlur(gray_frame, (21, 21), 0)
if first_frame is None:
first_frame = gray_frame_gau
delta_frame = cv2.absdiff(first_frame, gray_frame_gau)
thresh_frame = cv2.threshold(delta_frame, 65, 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) < 2000:
continue
x, y, w, h = cv2.boundingRect(contour)
rectangle = cv2.rectangle(frame, (x, y), (x+w, 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:]
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()
clean_thread.start()
cv2.imshow("Video", frame)
key = cv2.waitKey(1)
if key == ord('q'):
break
video.release()