python 视频转图像

# coding=utf-8

# 全局变量
VIDEO_PATH = 'd:/data/batball2020604.avi' # 视频地址
EXTRACT_FOLDER = 'd:/data/bat' # 存放帧图片的位置
EXTRACT_FREQUENCY = 2 # 帧提取频率


def extract_frames(video_path, dst_folder, index):
    # 主操作
    import cv2
    video = cv2.VideoCapture()
    if not video.open(video_path):
        print("can not open the video")
        exit(1)
    count = 1
    while True:
        _, frame = video.read()
        if frame is None:
            break
        if count % EXTRACT_FREQUENCY == 0:
            save_path = "{}/{:>05d}.jpg".format(dst_folder, index)
            rows, cols,c = frame.shape
            M = cv2.getRotationMatrix2D(((cols - 1) / 2.0, (rows - 1) / 2.0), 0, 1)
            dst = cv2.warpAffine(frame, M, (cols, rows))

            cv2.imwrite(save_path, dst)
            index += 1
        print('frame',count)
        count += 1
    video.release()
    #