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

use ConfigParser.get*() functions; move function definition out of main cycle; drop unused import; show recognition time

Browse source
This commit is contained in:
dmig 2018-11-18 13:49:16 +07:00
parent fa51ff7c10
commit 964e96377c
1 changed files with 21 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
src/cli/test.py
View file

@ -21,15 +21,18 @@ config.read(path + "/../config.ini")
video_capture = cv2.VideoCapture(config.get("video", "device_path")) video_capture = cv2.VideoCapture(config.get("video", "device_path"))
# Force MJPEG decoding if true # Force MJPEG decoding if true
if config.get("video", "force_mjpeg") == "true": if config.getboolean("video", "force_mjpeg"):
# Set a magic number, will enable MJPEG but is badly documented
video_capture.set(cv2.CAP_PROP_FOURCC, 1196444237) video_capture.set(cv2.CAP_PROP_FOURCC, 1196444237)
# Set the frame width and height if requested # Set the frame width and height if requested
if int(config.get("video", "frame_width")) != -1: fw = config.getint("video", "frame_width")
video_capture.set(cv2.CAP_PROP_FRAME_WIDTH, int(config.get("video", "frame_width"))) fh = config.getint("video", "frame_height")
if fw != -1:
video_capture.set(cv2.CAP_PROP_FRAME_WIDTH, fw)
if int(config.get("video", "frame_height")) != -1: if fh != -1:
video_capture.set(cv2.CAP_PROP_FRAME_HEIGHT, int(config.get("video", "frame_height"))) video_capture.set(cv2.CAP_PROP_FRAME_HEIGHT, fh)
# Let the user know what's up # Let the user know what's up
print(""" print("""
@ -47,6 +50,10 @@ def mouse(event, x, y, flags, param):
if event == cv2.EVENT_LBUTTONDOWN: if event == cv2.EVENT_LBUTTONDOWN:
slow_mode = not slow_mode slow_mode = not slow_mode
def print_text(line_number, text):
"""Print the status text by line number"""
cv2.putText(overlay, text, (10, height - 10 - (10 * line_number)), cv2.FONT_HERSHEY_SIMPLEX, .3, (0, 255, 0), 0, cv2.LINE_AA)
# Open the window and attach a a mouse listener # Open the window and attach a a mouse listener
cv2.namedWindow("Howdy Test") cv2.namedWindow("Howdy Test")
cv2.setMouseCallback("Howdy Test", mouse) cv2.setMouseCallback("Howdy Test", mouse)
@ -61,11 +68,13 @@ sec_frames = 0
fps = 0 fps = 0
# The current second we're counting # The current second we're counting
sec = int(time.time()) sec = int(time.time())
# recognition time
rec_tm = 0
# Wrap everything in an keyboard interupt handler # Wrap everything in an keyboard interupt handler
try: try:
while True: while True:
# Inclement the frames # Increment the frames
total_frames += 1 total_frames += 1
sec_frames += 1 sec_frames += 1
@ -109,14 +118,11 @@ try:
# Draw a stripe indicating the dark threshold # Draw a stripe indicating the dark threshold
cv2.rectangle(overlay, (8, 35), (20, 36), (255, 0, 0), thickness=cv2.FILLED) cv2.rectangle(overlay, (8, 35), (20, 36), (255, 0, 0), thickness=cv2.FILLED)
def print_text(line_number, text):
"""Print the status text by line number"""
cv2.putText(overlay, text, (10, height - 10 - (10 * line_number)), cv2.FONT_HERSHEY_SIMPLEX, .3, (0, 255, 0), 0, cv2.LINE_AA)
# Print the statis in the bottom left # Print the statis in the bottom left
print_text(0, "RESOLUTION: " + str(height) + "x" + str(width)) print_text(0, "RESOLUTION: %dx%d" % (height, width))
print_text(1, "FPS: " + str(fps)) print_text(1, "FPS: %d" % (fps, ))
print_text(2, "FRAMES: " + str(total_frames)) print_text(2, "FRAMES: %d" % (total_frames, ))
print_text(3, "RECOGNITION: %dms" % (round(rec_tm * 1000), ))
# Show that slow mode is on, if it's on # Show that slow mode is on, if it's on
if slow_mode: if slow_mode:
@ -130,8 +136,10 @@ try:
# SHow that this is an active frame # SHow that this is an active frame
cv2.putText(overlay, "SCAN FRAME", (width - 68, 16), cv2.FONT_HERSHEY_SIMPLEX, .3, (0, 255, 0), 0, cv2.LINE_AA) cv2.putText(overlay, "SCAN FRAME", (width - 68, 16), cv2.FONT_HERSHEY_SIMPLEX, .3, (0, 255, 0), 0, cv2.LINE_AA)
rec_tm = time.time()
# Get the locations of all faces and their locations # Get the locations of all faces and their locations
face_locations = face_recognition.face_locations(frame) face_locations = face_recognition.face_locations(frame)
rec_tm = time.time() - rec_tm
# Loop though all faces and paint a circle around them # Loop though all faces and paint a circle around them
for loc in face_locations: for loc in face_locations: