Python 批量抽取视频帧保存为图片

 1.导入所需函数库

主要使用OpenCV 和os

import cv2
import os

2.创建video类

class video():
  def __init__(self):
    self.start_name = '0000000.jpg' # 设置保存图片的起始命名
    self.timeF = 10 # 设置每隔10帧提取一张图片

 使用数字命名，方便大批量图片的存储及命名

这里也可以不使用类 直接创建视频处理函数

3.编写提取和保存图片帧的函数

def cut(self,video_path,save_path):
    
    filename = os.listdir(video_path) # 读取video_path 路径下的所有文件名
    n = 0   # 用于计数保存图片的数量
    
    for name in filename:  # 根据读取到的文件名读取视频文件
      name = video_path + name
      cv =  cv2.VideoCapture(name) # 读入视频文件，命名cv
      
      if cv.isOpened():  # 判断是否正常打开
        rval, frame = cv.read()  # rval 正常打开视频文件的标志位 frame为读取到的帧图片
        i = 1  # 用于计数每隔视频的帧数
      else:
        rval = False
        print('open video error!!')   # 打开文件出错时报错
        
      while rval:  # 正常打开 开始处理
        rval, frame = cv.read()
        jpg_name =save_path + str( int(self.start_name[0:-4]) + n).zfill(6) + '.jpg' # 命名保存的图片 
        
        if (i % self.timeF == 0):  # 每隔timeF帧进行存储操作
          n += 1
          try:   # 尝试存储图片文件 若出错则跳过 继续执行下一个
            cv2.imwrite(jpg_name, frame)  # 存储为图像
          except:
            pass
          # print(jpg_name)
        i += 1
      cv2.waitKey(1)
      cv.release()
      print(name+' done') # 提示某视频内容提取完成
    print('cut video done') # 提示全部内容提取完成

video_path,save_path分别是视频文件的路径和需要保存图片的路径

4.主函数编写

if __name__ == "__main__":

  video_path = 'video/'  # 设置为视频文件存储的路径即可
  img_path = 'img/'      

  if not os.path.exists(img_path): # 如果存储图片的文件夹不存在，自动创建保存图片文件夹
    os.makedirs(img_path)
  v = video()          
  v.cut(video_path,img_path)  # 执行提取图片程序

5.完整代码

import cv2
import os


class video():
  def __init__(self):
    self.start_name = '0000000.jpg'
    self.timeF = 10
    
  def cut(self,video_path,save_path):
    
    filename = os.listdir(video_path)
    n = 0
    
    for name in filename:
      name = video_path + name
      cv =  cv2.VideoCapture(name) # 读入视频文件，命名cv
      
      if cv.isOpened():  # 判断是否正常打开
        rval, frame = cv.read()
        i = 1
      else:
        rval = False
        print('open video error!!')
        
      while rval:  # 正常打开 开始处理
        rval, frame = cv.read()
        jpg_name =save_path + str( int(self.start_name[0:-4]) + n).zfill(6) + '.jpg' # 命名保存的图片 
        
        if (i % self.timeF == 0):  # 每隔timeF帧进行存储操作
          n += 1
          try:
            cv2.imwrite(jpg_name, frame)  # 存储为图像
          except:
            pass
          # print(jpg_name)
        i += 1
      cv2.waitKey(1)
      cv.release()
      print(name+' done')
    print('cut video done')


if __name__ == "__main__":

  video_path = 'video/'  # 设置为视频文件存储的路径即可
  img_path = 'img/'      

  if not os.path.exists(img_path): # 如果存储图片的文件夹不存在，自动创建保存图片文件夹
    os.makedirs(img_path) 
  v = video()         
  v.cut(video_path,img_path)  # 执行提取图片程序