人脸搜索搭建

import face_recognition
import cv2
import numpy as np

# 加载已知人脸图像和对应的名称
known_image = face_recognition.load_image_file("known_face.jpg")
known_encoding = face_recognition.face_encodings(known_image)[0]
known_names = ["Known Person"]

# 初始化视频捕获
video_capture = cv2.VideoCapture(0)

while True:
    # 获取当前帧
    ret, frame = video_capture.read()
    small_frame = cv2.resize(frame, (0, 0), fx=0.25, fy=0.25)
    rgb_small_frame = small_frame[:, :, ::-1]

    # 查找当前帧中所有人脸的编码
    face_encodings = face_recognition.face_encodings(rgb_small_frame)

    for face_encoding in face_encodings:
        # 比较当前人脸编码与已知人脸编码
        matches = face_recognition.compare_faces([known_encoding], face_encoding)
        name = "Unknown"

        # 如果找到匹配的人脸
        if True in matches:
            first_match_index = matches.index(True)
            name = known_names[first_match_index]

        # 在帧上绘制姓名
        for (top, right, bottom, left) in face_locations:
            cv2.rectangle(frame, (left*4, top*4), (right*4, bottom*4), (0, 0, 255), 2)
            cv2.putText(frame, name, (left*4 + 6, bottom*4 - 6), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (255, 255, 255), 1)

    # 显示结果帧
    cv2.imshow('Video', frame)

    # 按q退出循环
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

video_capture.release()
cv2.destroyAllWindows()