通过图片识别名人

sh复制pip install face_recognition opencv-python dlib

复制dataset/
    ├── person1/
    │   ├── image1.jpg
    │   ├── image2.jpg
    ├── person2/
    │   ├── image1.jpg
    │   ├── image2.jpg

python复制import face_recognition
import cv2
import os
import numpy as np

# 加载名人图片并生成编码
def load_known_faces(known_faces_dir):
    known_faces = []
    known_names = []

    for name in os.listdir(known_faces_dir):
        person_dir = os.path.join(known_faces_dir, name)
        if not os.path.isdir(person_dir):
            continue

        for filename in os.listdir(person_dir):
            filepath = os.path.join(person_dir, filename)
            image = face_recognition.load_image_file(filepath)
            encodings = face_recognition.face_encodings(image)

            if len(encodings) > 0:
                known_faces.append(encodings[0])
                known_names.append(name)

    return known_faces, known_names

# 识别图片中的名人
def recognize_faces(image_path, known_faces, known_names):
    image = face_recognition.load_image_file(image_path)
    face_locations = face_recognition.face_locations(image)
    face_encodings = face_recognition.face_encodings(image, face_locations)

    recognized_names = []

    for face_encoding in face_encodings:
        matches = face_recognition.compare_faces(known_faces, face_encoding)
        name = "Unknown"

        face_distances = face_recognition.face_distance(known_faces, face_encoding)
        best_match_index = np.argmin(face_distances)
        if matches[best_match_index]:
            name = known_names[best_match_index]

        recognized_names.append(name)

    return recognized_names, face_locations

# 主函数
def main():
    known_faces_dir = "dataset"
    test_image_path = "test_image.jpg"

    known_faces, known_names = load_known_faces(known_faces_dir)
    recognized_names, face_locations = recognize_faces(test_image_path, known_faces, known_names)

    print("Recognized faces:", recognized_names)

    # 显示结果
    image = cv2.imread(test_image_path)
    for (top, right, bottom, left), name in zip(face_locations, recognized_names):
        cv2.rectangle(image, (left, top), (right, bottom), (0, 255, 0), 2)
        cv2.putText(image, name, (left, top - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)

    cv2.imshow("Image", image)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

if __name__ == "__main__":
    main()