def separate_key_frames_from_row()

in video_processing/modules/frames.py [0:0]


def separate_key_frames_from_row(path: Path, row: dict[str, list]):
    key_frames = [Image.open(path.joinpath(key_frame)) for key_frame in row["frames"]]
    first = key_frames[0]
    mid = None
    last = None
    if len(key_frames) == 2:
        last = key_frames[1]
    elif len(key_frames) > 2:
        mid = key_frames[len(key_frames) // 2]
        last = key_frames[-1]
    return key_frames, first, mid, last