frames_extraction.py

import cv2
from os import makedirs
from os.path import join, exists
import glob

training_videos_folder = ["train/0", "train/1"]

for folder in training_videos_folder:
    videos_path = glob.glob(join(folder, "*.mp4"))
    folder = folder.split("/")[1]

    counter = 0
    for video_path in videos_path:
        cap = cv2.VideoCapture(video_path)
        vid = video_path.split("/")[-1]
        vid = vid.split(".")[0]
        frameRate = cap.get(5)  # frame rate

        if not exists("train_frames/" + folder + "/video_" + str(int(counter))):
            makedirs("train_frames/" + folder + "/video_" + str(int(counter)))

        while cap.isOpened():
            frameId = cap.get(1)  # current frame number
            ret, frame = cap.read()
            if not ret:
                break

            filename = (
                "train_frames/"
                + folder
                + "/video_"
                + str(int(counter))
                + "/image_"
                + str(int(frameId) + 1)
                + ".jpg"
            )
            cv2.imwrite(filename, frame)

        cap.release()

        if counter % 100 == 0:
            print(counter, "Done....")
        counter += 1