自学教程：opencv读取视频并保存图像的方法

import cv2import os

video_path = './未戴安全帽视频01/'image_path = './images/'

video_files = [i for i in os.listdir(video_path) if i.split('.')[-1] in ['mp4']]len(video_files)

#　video_file:'./未戴安全帽视频01/中建四局-东围墙5_001_2021-03-22-18-04-28_2021-03-22-18-04-33.mp4', # pic_dir:'中建四局-东围墙5_001_2021-03-22-18-04-28_2021-03-22-18-04-33'def get_image(video_file, pic_dir):    if not os.path.exists(pic_dir):        os.makedirs(pic_dir)        # cv2读取视频文件    vc = cv2.VideoCapture(video_file)    index = 0    # 判断载入的视频是否可以打开    rval = vc.isOpened()    while rval:  # 循环读取视频帧        index = index + 1                rval, frame = vc.read()        # 每十帧保存一张图片        if index * 10 % 1 == 0:            if rval:                # cv2.imshow("capture", frame)                save_file = pic_dir + str(index).zfill(5) + '.png'                cv2.imwrite(save_file, frame)  # 存储为图像，保存名为文件夹名                cv2.waitKey(1)            else:                break        vc.release()    print("已保存%d" %(index - 1) + "张图片")        # video_file = './未戴安全帽视频01/01.mp4'# pic_path = '01/'# get_image(video_file, image_path + pic_path)

for file in video_files:    video_file = video_path + file    pic_path = image_path + file.replace('.mp4', '/')    get_image(video_file, pic_path)

import cv2import osdef save_img():    video_path = r'F:/test/3.10'    videos = os.listdir(video_path)    for video_name in videos:        file_name = video_name.split('.')[0]        folder_name = video_path +'_'+ file_name        os.makedirs(folder_name, exist_ok=True)        print(video_path + '/' + video_name)        vc = cv2.VideoCapture(video_path + '/' + video_name)        # 读入视频文件        c = 0        rval = vc.isOpened()        while rval:  # 循环读取视频帧            c = c + 1            rval, frame = vc.read()            if c%10 ==0:                pic_path = folder_name + '/'                if rval:                    cv2.imwrite(pic_path + str(c) + '.png', frame)  # 存储为图像，保存名为文件夹名                    cv2.waitKey(1)                else:                    break        vc.release()        print('save_success')        print(folder_name)save_img()

if not os.path.exists(path):    os.makedirs(path)