OpenCV множественное распознавание лиц

Question

Мой следующий код использует данное видео с одним человеком и возвращает видео, которое содержит только лицо человека в видео. Я пытаюсь заставить его работать на видео с большим количеством людей (создать видео для каждого лица в оригинальном видео).

# import libraries
import cv2
import face_recognition

# Get a reference to webcam
video_capture = cv2.VideoCapture("media5.mp4")

fourcc = cv2.VideoWriter_fourcc(*'mp4v')
shape = 50, 50
output = cv2.VideoWriter('output.mp4', fourcc, 20, shape)

# Initialize variables
face_locations = []

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

    if str(type(frame)) == "<class 'NoneType'>":
        break;
    # Convert the image from BGR color (which OpenCV uses) to RGB color (which face_recognition uses)
    rgb_frame = frame[:, :, ::-1]

    # Find all the faces in the current frame of video
    face_locations = face_recognition.face_locations(rgb_frame)

    # Display the results
    for top, right, bottom, left in face_locations:
        # Draw a box around the face
        crop_img = frame[top:bottom, left:right]
        resized = cv2.resize(crop_img, shape)
        output.write(resized)
        cv2.imshow("cropped", crop_img)
        cv2.rectangle(frame, (left, top), (right, bottom), (0, 0, 255), 2)
    # 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()
output.release()
cv2.destroyAllWindows()

Любые советы?