python视频切片脚本

import cv2
import glob

def my_task(video_capdir, ):
    name = video_capdir.split('\\')[-1]
    print(name[:-4])
    video_cap = cv2.VideoCapture(video_capdir)
    # print(video_cap.get(cv2.CAP_PROP_POS_FRAMES))  # 查看当前的帧数
    success, image = video_cap.read()  # 获取第一帧
    count = 0
    while success:
        if count % 40 == 0: # 图片截取间隔
            cv2.imwrite(f'dataset/mfsd/0/{name[:-4]}{count:05d}th.jpg', image)
        success, image = video_cap.read()  # 每次读取一帧
        count += 1


all_vdir = glob.glob('my_dir/*.mp4')
for each in all_vdir:
    my_task(video_capdir=each)