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

Made displaying video optional

This commit is contained in:
slashtechno 2022-12-17 16:09:39 -06:00
parent 1c6613b206
commit 5c87752526
Signed by: slashtechno
GPG Key ID: 8EC1D9D9286C2B17
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@ -15,6 +15,7 @@ VIEW_SCALE = os.getenv("VIEW_SCALE")
DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S" DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
# RUN_SCALE = 0.25 # RUN_SCALE = 0.25
# VIEW_SCALE = 0.75 # VIEW_SCALE = 0.75
DISPLAY = False
def find_face_from_name(name): def find_face_from_name(name):
@ -40,12 +41,13 @@ process_this_frame = True
# Load the config file, if it does not exist or is blank, create it # Load the config file, if it does not exist or is blank, create it
config = { config = {
"URL": "rtsp://localhost:8554/example", "URL": "rtsp://localhost:8554/example",
"RUN_SCALE": "0.25", "run_scale": "0.25",
"VIEW_SCALE": "0.75", "view_scale": "0.75",
"faces": { "faces": {
"example1": {"image": "example1.jpg", "last_seen": ""}, "example1": {"image": "example1.jpg", "last_seen": ""},
"example2": {"image": "example2.jpg", "last_seen": ""}, "example2": {"image": "example2.jpg", "last_seen": ""},
}, },
"display": True
} }
config_path = pathlib.Path("config.json") config_path = pathlib.Path("config.json")
if config_path.exists(): if config_path.exists():
@ -70,6 +72,12 @@ if VIEW_SCALE:
config["VIEW_SCALE"] = VIEW_SCALE config["VIEW_SCALE"] = VIEW_SCALE
else: else:
VIEW_SCALE = float(config["VIEW_SCALE"]) VIEW_SCALE = float(config["VIEW_SCALE"])
if DISPLAY:
config["DISPLAY"] = DISPLAY
else:
DISPLAY = config["display"]
print(f"Current config: {config}")
for face in config["faces"]: for face in config["faces"]:
# Load a sample picture and learn how to recognize it. # Load a sample picture and learn how to recognize it.
@ -165,7 +173,8 @@ while True:
view_frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1 view_frame, name, (left + 6, bottom - 6), font, 1.0, (255, 255, 255), 1
) )
# Display the resulting image # Display the resulting image if DISPLAY is set to true
if DISPLAY:
cv2.imshow("Scaled View", view_frame) cv2.imshow("Scaled View", view_frame)
# Hit 'q' on the keyboard to quit! # Hit 'q' on the keyboard to quit!