slashtechno
/
wyzely-detect
mirror of https://github.com/slashtechno/wyzely-detect
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed rapid notifications

This commit is contained in:
slashtechno 2022-12-18 13:44:42 -06:00
parent fed6f03c7f
commit 717c52d14e
Signed by: slashtechno
GPG Key ID: 8EC1D9D9286C2B17
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.py
View File

@ -124,7 +124,9 @@ while True:
rgb_run_frame = run_frame[:, :, ::-1] rgb_run_frame = run_frame[:, :, ::-1]
# Find all the faces and face encodings in the current frame of video # Find all the faces and face encodings in the current frame of video
# model cnn is gpu accelerated, but hog is cpu only # model cnn is gpu accelerated, but hog is cpu only
face_locations = face_recognition.face_locations(rgb_run_frame, model="hog") # This crashes the program without output on my laptop when it's running without Docker compose face_locations = face_recognition.face_locations(
rgb_run_frame, model="hog"
) # This crashes the program without output on my laptop when it's running without Docker compose
face_encodings = face_recognition.face_encodings(rgb_run_frame, face_locations) face_encodings = face_recognition.face_encodings(rgb_run_frame, face_locations)
face_names = [] face_names = []
for face_encoding in face_encodings: for face_encoding in face_encodings:
@ -142,7 +144,7 @@ while True:
# If it's never been seen, set the last seen time to x+5 seconds ago so it will be seen # If it's never been seen, set the last seen time to x+5 seconds ago so it will be seen
# Kind of a hacky way to do it, but it works... hopefully # Kind of a hacky way to do it, but it works... hopefully
if last_seen == "": if last_seen == "":
print(f"{name} has been seen") print(f"{name} has been seen for the first time")
config["faces"][name]["last_seen"] = ( config["faces"][name]["last_seen"] = (
datetime.datetime.now() - datetime.timedelta(seconds=15) datetime.datetime.now() - datetime.timedelta(seconds=15)
).strftime(DATETIME_FORMAT) ).strftime(DATETIME_FORMAT)
@ -152,10 +154,6 @@ while True:
last_seen, DATETIME_FORMAT last_seen, DATETIME_FORMAT
) > datetime.timedelta(seconds=10): ) > datetime.timedelta(seconds=10):
print(f"{name} has been seen") print(f"{name} has been seen")
# Update the last seen time
config["faces"][name]["last_seen"] = datetime.datetime.now().strftime(
DATETIME_FORMAT
)
# Send a notification # Send a notification
print(f"Sending notification to{NTFY_URL}") print(f"Sending notification to{NTFY_URL}")
requests.post( requests.post(
@ -168,6 +166,10 @@ while True:
}, },
) )
print("Writing config...") print("Writing config...")
# Update the last seen time
config["faces"][name]["last_seen"] = datetime.datetime.now().strftime(
DATETIME_FORMAT
)
write_config() write_config()
face_names.append(name) face_names.append(name)
# Display the results # Display the results