video2frame.py

import os
import cv2
import glob
from pathlib import Path


FOLDER_VIDEOS = "<<PATH>>/*.mp4"
FOLDER_FRAMES = "<<PATH>>"

for i, video in enumerate(glob.glob(FOLDER_VIDEOS)):
    Path(os.path.join(FOLDER_FRAMES, f"{i}")).mkdir(parents=True, exist_ok=True)

    print(video, i)

    count_frame = 0
    cap = cv2.VideoCapture(video)
    fps = cap.get(cv2.CAP_PROP_FPS)

    dets_net = []

    while(cap.isOpened()):

        ret, frame_read = cap.read()
        elapsed_video_seconds = count_frame/fps

        count_frame = count_frame + 1

        # To skip some frames
        # if count_frame % 10000 != 0:
        #     continue
            
        if ret == True:
            frame_rgb = cv2.cvtColor(frame_read, cv2.COLOR_BGR2RGB)
            cv2.imwrite(os.path.join(FOLDER_FRAMES, f"{i}",  f"{count_frame}.png"), frame_read)
        else:
            break
    cap.release()