split video into jpegs. ビデオをフレーム画像に分割する

video_into_jpegs.py

import cv2
import os

def save_all_frames(video_path, dir_path, basename, ext='jpg'):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        return

    os.makedirs(dir_path, exist_ok=True)
    base_path = os.path.join(dir_path, basename)

    digit = len(str(int(cap.get(cv2.CAP_PROP_FRAME_COUNT))))

    n = 0

    while True:
        ret, frame = cap.read()
        if ret:
            print('frame num', n)
            cv2.imwrite('{}_{}.{}'.format(base_path, str(n).zfill(digit), ext), frame)
            n += 1
        else:
            return
          
def save_frames(video_path, dir_path, basename, ext='jpg', maxFrame = 60):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        return

    os.makedirs(dir_path, exist_ok=True)
    base_path = os.path.join(dir_path, basename)

    digit = len(str(int(cap.get(cv2.CAP_PROP_FRAME_COUNT))))

    n = 0
    while True:
        ret, frame = cap.read()
        if ret:
            if (n < maxFrame):
                cv2.imwrite('{}_{}.{}'.format(base_path, str(n).zfill(digit), ext), frame)
            n += 1
        else:
            return

def save_part_frames(video_path, dir_path, basename, ext='jpg', start = 10, end = 70, orderFlip = False):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        return

    os.makedirs(dir_path, exist_ok=True)
    base_path = os.path.join(dir_path, basename)

    digit = len(str(int(cap.get(cv2.CAP_PROP_FRAME_COUNT))))

    n = 0
    while True:
        ret, frame = cap.read()
        if ret:
            
            if (start <= n and  n < end):
                if(orderFlip):
                    cv2.imwrite('{}_{}.{}'.format(base_path, str(n-start).zfill(digit), ext), frame)
                else:
                    _i = (end -start -1) - (n-start)
                    cv2.imwrite('{}_{}.{}'.format(base_path, str(_i).zfill(digit), ext), frame)
            n += 1
        else:
            return