批量将视频转为图片并以视频名称为文件夹保存

# 导入所需要的库
import cv2
import os
import glob
import numpy as np
  
# 定义保存图片函数
# image:要保存的图片名字
# addr；图片地址与相片名字的前部分
# num: 相片，名字的后缀。int 类型
def save_image(image,addr,num):
    address = addr + str(num)+ '.jpg'
    cv2.imwrite(address,image)
seq_dir = os.path.expanduser('./')#视频目录
img_files=glob.glob(seq_dir + '*.avi') 
for f, img_file in enumerate(img_files):
	videoCapture = cv2.VideoCapture(img_file)
	success, frame = videoCapture.read()
	i = 0
	while success:
		i = i + 1
		try:
			add=os.mkdir(os.path.join('./guanjianzhen/', str(img_file)))#图片输出目录
		except Exception as error:
			pass		
		save_image(frame,'./guanjianzhen/'+str(img_file)+'/image',i)
		if success:
			print('save image:',i)
		success, frame = videoCapture.read()