python工具-將視頻按幀截取圖片（附代碼）

描述：將一個視頻流按幀數截取大量的圖片

用途：AI的數據集制作，得到大量的圖片，之后將其打標簽

更改的地方

1.default--間隔的幀數   2.input/output--輸入視頻的路徑、存放截取圖片的路徑 （將路徑放入后面的‘ ’中即可）前面加r可表示絕對路徑 eg:

args = parser.parse_args(['--input',r'F:\data_video\IMG_4395.MOV','--output',r'F:data_rgb_pic\7video'])

直接上代碼

import cv2
import argparse
import os
def parse_args():
    """
    Parse input arguments
    """
    parser = argparse.ArgumentParser(description='Process pic')
    parser.add_argument('--input', help='video to process', dest='input', default=None, type=str)
    parser.add_argument('--output', help='pic to store', dest='output', default=None, type=str)
    #default為間隔多少幀截取一張圖片
    parser.add_argument('--skip_frame', dest='skip_frame', help='skip number of video', default=100, type=int)
    #input為輸入視頻的路徑 ，output為輸出存放圖片的路徑
    args = parser.parse_args(['--input','','--output',''])
    return args
 
def process_video(i_video, o_video, num):
    cap = cv2.VideoCapture(i_video)
    num_frame = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    expand_name = '.jpg'
    if not cap.isOpened():
        print("Please check the path.")
    cnt = 0
    count = 0
    while 1:
        ret, frame = cap.read()
        cnt += 1
        #  how
        # many
        # frame
        # to
        # cut
        if cnt % num == 0:
            count += 1
            cv2.imwrite(os.path.join(o_video, str(count) + expand_name), frame)
 
        if not ret:
            break
 
if __name__ == '__main__':
    args = parse_args()
    if not os.path.exists(args.output):
        os.makedirs(args.output)
    print('Called with args:')
    print(args)
    process_video(args.input, args.output, args.skip_frame)

參考

https://blog.csdn.net/qq_36190978/article/details/85284484