opencv读取视频并保存图像的方法

import cv2
import 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 cv2
import os

def 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)