Skip to content

Commit 3696ed5

Browse files
committedJun 29, 2018
info logs in face recognizer; tuned accuracy thresholds
1 parent abc29ee commit 3696ed5

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed
 

‎face_recognition.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,9 @@
2929
import cv2 as cv
3030
import pickle
3131

32+
from logger import get_logger
33+
log = get_logger()
34+
3235

3336
class FaceRecognition:
3437

@@ -42,6 +45,7 @@ def __init__(self, face_cascade_xml, eye_cascade_xml):
4245
eye_cascade = cv.CascadeClassifier(eye_cascade_xml)
4346

4447
# creating a face recognier with pretrained data
48+
log.info("creating recognier from pre-trained data")
4549
recognizer = cv.face.LBPHFaceRecognizer_create()
4650
recognizer.read("dataset/face_trainer.yml")
4751

@@ -59,6 +63,7 @@ def __init__(self, face_cascade_xml, eye_cascade_xml):
5963
org_labels = pickle.load(f)
6064
labels = {v:k for k,v in org_labels.items()}
6165

66+
log.info("capturing video data")
6267
while True:
6368

6469
ret, img = cap.read()
@@ -81,7 +86,7 @@ def __init__(self, face_cascade_xml, eye_cascade_xml):
8186
# identify the face with recognizer
8287
index, conf = recognizer.predict(roi_gray)
8388

84-
if conf > 45 and conf <= 85:
89+
if conf > 75 and conf <= 95:
8590
name = labels[index]
8691
# Hurray, we detected a face !!!
8792
print("Identified face: Name: %s, index: %d, confidence level: %d" % (name, index, conf))

0 commit comments

Comments
 (0)
Please sign in to comment.