danvics/howdy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Starting compare camera earlier and printing a warning if too many models are being added

Browse source
This commit is contained in:
boltgolt 2018-02-20 22:18:29 +01:00
parent edfb6ad7e2
commit 3d6d0e6c1d
2 changed files with 26 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cli/add.py
View file

@ -45,6 +45,11 @@ try:
except FileNotFoundError: except FileNotFoundError:
encodings = [] encodings = []
# Print a warning if too many encodings are being added
if len(encodings) > 2:
print("WARNING: Every additional model slows down the face recognition engine")
print("Press ctrl+C to cancel")
print("Adding face model for the user account " + user) print("Adding face model for the user account " + user)
# Set the default label # Set the default label

31
compare.py
View file

@ -1,18 +1,20 @@
# Compare incomming video with known faces # Compare incomming video with known faces
# Running in a local python instance to get around PATH issues # Running in a local python instance to get around PATH issues
# Import time so we can start timing asap
import time
# Start timing
timings = [time.time()]
# Import required modules # Import required modules
import cv2 import cv2
import sys import sys
import os import os
import json import json
import time
import math import math
import configparser import configparser
# Start timing
timings = [time.time()]
# Read config from disk # Read config from disk
config = configparser.ConfigParser() config = configparser.ConfigParser()
config.read(os.path.dirname(os.path.abspath(__file__)) + "/config.ini") config.read(os.path.dirname(os.path.abspath(__file__)) + "/config.ini")
@ -54,13 +56,21 @@ if len(models) < 1:
for model in models: for model in models:
encodings += model["data"] encodings += model["data"]
# Import face recognition, takes some time # Add the time needed to start the script
timings.append(time.time())
import face_recognition
timings.append(time.time()) timings.append(time.time())
# Start video capture on the IR camera # Start video capture on the IR camera
video_capture = cv2.VideoCapture(int(config.get("video", "device_id"))) video_capture = cv2.VideoCapture(int(config.get("video", "device_id")))
# Capture a single frame so the camera becomes active
# This will let the camera adjust its light levels while we're importing for faster scanning
video_capture.read()
# Note the time it took to open the camera
timings.append(time.time())
# Import face recognition, takes some time
import face_recognition
timings.append(time.time()) timings.append(time.time())
# Fetch the max frame height # Fetch the max frame height
@ -126,15 +136,16 @@ while True:
print("Time spend") print("Time spend")
print_timing("Starting up", 0) print_timing("Starting up", 0)
print_timing("Importing face_recognition", 1) print_timing("Opening the camera", 1)
print_timing("Opening the camera", 2) print_timing("Importing face_recognition", 2)
print_timing("Searching for known face", 3) print_timing("Searching for known face", 3)
print("\nResolution") print("\nResolution")
print(" Native: " + str(height) + "x" + str(width)) print(" Native: " + str(height) + "x" + str(width))
print(" Used: " + str(scale_height) + "x" + str(scale_width)) print(" Used: " + str(scale_height) + "x" + str(scale_width))
print("\nFrames searched: " + str(frames) + " (" + str(round(float(frames) / (timings[4] - timings[2]), 2)) + " fps)") # Show the total number of frames and calculate the FPS by deviding it by the total scan time
print("\nFrames searched: " + str(frames) + " (" + str(round(float(frames) / (timings[4] - timings[3]), 2)) + " fps)")
print("Dark frames ignored: " + str(dark_tries)) print("Dark frames ignored: " + str(dark_tries))
print("Certainty of winning frame: " + str(round(match * 10, 3))) print("Certainty of winning frame: " + str(round(match * 10, 3)))