root_face目录结构:
人名:
人脸照片
import face_recognition
import cv2
import os
import numpy as np
def face_encode(img, model='large'):
points = face_recognition.face_locations(img, model='cnn')
encodeing = face_recognition.face_encodings(img, points, model=model)
return points, encodeing
path = r'./root_face'
dirs = os.listdir(path)
face_dict = {}
root_face = []
index = 0
for dir in dirs:
dir_path = os.path.join(path, dir)
files = os.listdir(dir_path)
for file in files:
file_path = os.path.join(dir_path, file)
img = cv2.imread(file_path)
w, h = img.shape[1::-1]
ratio = w / h
h = 400
img = cv2.resize(img, (int(h * ratio), h))
_, encodeing = face_encode(img)
face_dict['{}'.format(index)] = '{}'.format(dir)
root_face.append(encodeing[0])
index += 1
def gstreamer_pipeline(
capture_width=1280,
capture_height=720,
display_width=1280,
display_height=720,
framerate=24,
flip_method=0,
):
return (
"nvarguscamerasrc ! "
"video/x-raw(memory:NVMM), "
"width=(int)%d, height=(int)%d, "
"format=(string)NV12, framerate=(fraction)%d/1 ! "
"nvvidconv flip-method=%d ! "
"video/x-raw, width=(int)%d, height=(int)%d, format=(string)BGRx ! "
"videoconvert ! "
"video/x-raw, format=(string)BGR ! appsink"
% (
capture_width,
capture_height,
framerate,
flip_method,
display_width,
display_height,
)
)
def show_camera():
cap = cv2.VideoCapture(gstreamer_pipeline(flip_method=0), cv2.CAP_GSTREAMER)
while cap.isOpened():
flag, frame = cap.read()
frame_count = cap.get(cv2.CAP_PROP_FPS)
print(frame_count)
w, h = frame.shape[1::-1]
ratio = w / h
h = 150
frame = cv2.resize(frame, (int(h * ratio), h))
points, encodeings = face_encode(frame)
frame_count = cap.get(cv2.CAP_PROP_FPS)
print(frame_count)
for pts, encodeing in zip(points, encodeings):
cv2.rectangle(frame, tuple((pts[1], pts[0])), tuple((pts[3], pts[2])), (0, 255, 0))
face_distances = face_recognition.face_distance(root_face, encodeing)
if np.min(face_distances) < 0.4:
continue
best_match_index = np.argmin(face_distances)
str1 = face_dict['{}'.format(best_match_index)]
cv2.putText(frame, str1, tuple((pts[1], pts[0])), cv2.FONT_HERSHEY_SIMPLEX, 1.2, (255, 255, 255), 2)
cv2.imshow("windowName", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
k = cv2.waitKey()
if k == 27:
break
if cv2.getWindowProperty('windowName', cv2.WND_PROP_AUTOSIZE) < 1:
break
cap.release()
cv2.destroyAllWindows()
if __name__ == "__main__":
show_camera()