如何用Python源码识别视频人物角色 Python源码结合图像识别工具使用

pip install face_recognition

import face_recognition
import cv2

# 加载已知人物照片和姓名
known_face_encodings = []
known_face_names = []

# 假设你有三个人物：Alice, Bob, Charlie
# 加载Alice的照片
alice_image = face_recognition.load_image_file("alice.jpg")
alice_face_encoding = face_recognition.face_encodings(alice_image)[0] # 假设照片中只有一张人脸
known_face_encodings.append(alice_face_encoding)
known_face_names.append("Alice")

# 加载Bob的照片
bob_image = face_recognition.load_image_file("bob.jpg")
bob_face_encoding = face_recognition.face_encodings(bob_image)[0]
known_face_encodings.append(bob_face_encoding)
known_face_names.append("Bob")

# 加载Charlie的照片
charlie_image = face_recognition.load_image_file("charlie.jpg")
charlie_face_encoding = face_recognition.face_encodings(charlie_image)[0]
known_face_encodings.append(charlie_face_encoding)
known_face_names.append("Charlie")


# 加载视频
video_capture = cv2.VideoCapture("your_video.mp4")

while(True):
    # Grab a single frame of video
    ret, frame = video_capture.read()

    if not ret:
        break # 视频结束

    # Find all the faces and face encodings in the current frame of video
    face_locations = face_recognition.face_locations(frame)
    face_encodings = face_recognition.face_encodings(frame, face_locations)

    # Loop through each face found in the frame
    for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
        # See if the face is a match for the known face(s)
        matches = face_recognition.compare_faces(known_face_encodings, face_encoding)

        name = "Unknown"

        # If a match was found in known_face_encodings, just use the first one.
        if True in matches:
            first_match_index = matches.index(True)
            name = known_face_names[first_match_index]

        # Draw a box around the face
        cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)

        # Draw a label below the face
        cv2.rectangle(frame, (left, bottom - 35), (right, bottom), (0, 0, 255), cv2.FILLED)
        font = cv2.FONT_HERSHEY_DUPLEX
        cv2.putText(frame, name, (left + 6, bottom - 6), font, 0.8, (255, 255, 255), 1)

    # Display the resulting image
    cv2.imshow('Video', frame)

    # Hit 'q' on the keyboard to quit!
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# Release handle to the webcam
video_capture.release()
cv2.destroyAllWindows()