python视频批量抽帧保存代码

为了获取不同飞机的数据以供后续处理，我在这篇博客python opencv视频抽帧保存的基础上写了将从不同视频抽取的帧保存到不同文件夹的的代码，且每张图像的文件名不会重复

import cv2
import os
import time

# 视频路径
video_path = r'G:\PlaneData\planevideos'
# 保存图片的路径
savedpath = r'G:\PlaneData\planeframes'

video_list = os.listdir(video_path)

# 保存图片的帧率间隔
count = 10
i = 0



for index, video_name in enumerate(video_list):
    j = 0
    video_path_ = os.path.join(video_path, video_name)
    # 某个视频帧保存路径
    save_path_ = os.path.join(savedpath, video_name.split('.')[0])
    if os.path.exists(save_path_):
        pass
    else:
        os.mkdir(save_path_)

    # 开始读视频

    videoCapture = cv2.VideoCapture(video_path_)

    print("正在处理第{}个视频，总共{}个视频".format(index + 1, len(video_list)))

    while True:

        success, frame = videoCapture.read()

        i += 1

        if (i % count == 0):
            # 保存图片

            j += 1

            savedname = video_name[:-4] + str(i).zfill(4) + '.jpg'

            cv2.imwrite(os.path.join(save_path_, savedname), frame)

            print('image of %s is saved' % (savedname))

        if not success:
            print('video is all read')

            break

    videoCapture.release()

    time.sleep(5)