jetson nano face_recognition 人脸识别代码

path中存放人脸数据 并以名字命名

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'    
files = os.listdir(path)
face_dict = {}
root_face = []
index = 0
for file in files:
    file_path = os.path.join(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(file.split('.')[0])
    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()