got_body.py

def video_parser(video_name = "videos/{}".format("vid1.mp4")):
    video_capture = cv2.VideoCapture(video_name)
    OFFSET = 50
    OFFSET_TOP = 100
    QUALITY = 0.25
    CLASSES = {
        "lannister": ['tyrion', 'cersei', 'jaime', 'joffery', 'myrcella', 'tommen', 'tywin'],

    #-----------------------

        "stark": ['arya', 'brandon', 'eddard', 'robb', 'sansa']
    }
    
    known_face_encodings = []
    # [2] ADD THAT NAME HERE ALSO
    known_face_names = [
        "tyrion",
        'cersei', 
        'jaime', 
        'joffery', 
        'myrcella', 
        'tommen', 
        'tywin',
        
        "arya",
        'arya', 
        'brandon', 
        'eddard', 
        'robb', 
        'sansa'
    ]
    # [3] ADD IMAGE WITH NAME.JPG FORMAT TO FACES/ FOLDER

    # mapping
    for name in known_face_names:
        temp_image = face_recognition.load_image_file("faces/{}.jpg".format(name))
        temp_encoding = face_recognition.face_encodings(temp_image)[0]
        known_face_encodings.append(temp_encoding)
    
    # Initialize some variables
    face_locations = []
    face_encodings = []
    face_names = []
    process_this_frame = True
    COUNT = 0