视频（avi）转换为图片（Python代码实现）

下面Python代码是把视频格式是avi的视频转换为一帧一帧的图片

  

   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      video_src_src_path = 'data/hmdb_data_demo' #数据集路径
     
    
   

    

     

    
    

     

      label_name = os.listdir(video_src_src_path)
     
    
   

    

     

    
    

     

      label_dir = {}
     
    
   

    

     

    
    

     

      index = 0
     
    
   

    

     

    
    

     

      for i in label_name:
     
    
   

    

     

    
    

     
    if i.startswith('.'):
     
    
   

    

     

    
    

     
        continue
     
    
   

    

     

    
    

     

          label_dir[i] = index
     
    
   

    

     

    
    

     

          index += 1
     
    
   

    

     

    
    

     

          video_src_path = os.path.join(video_src_src_path, i)
     
    
   

    

     

    
    

     

          video_save_path = os.path.join(video_src_src_path, i) + '_jpg'
     
    
   

    

     

    
    

     
    if not os.path.exists(video_save_path):
     
    
   

    

     

    
    

     

              os.mkdir(video_save_path)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          videos = os.listdir(video_src_path)
     
    
   

    

     

    
    

     
    # 过滤出avi文件
     
    
   

    

     

    
    

     

          videos = filter(lambda x: x.endswith('avi'), videos)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    for each_video in videos:
     
    
   

    

     

    
    

     

              each_video_name, _ = each_video.split('.')
     
    
   

    

     

    
    

     
        if not os.path.exists(video_save_path + '/' + each_video_name):
     
    
   

    

     

    
    

     

                  os.mkdir(video_save_path + '/' + each_video_name)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              each_video_save_full_path = os.path.join(video_save_path, each_video_name) + '/'
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              each_video_full_path = os.path.join(video_src_path, each_video)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

              cap = cv2.VideoCapture(each_video_full_path)
     
    
   

    

     

    
    

     

              frame_count = 1
     
    
   

    

     

    
    

     

              success = True
     
    
   

    

     

    
    

     
        while success:
     
    
   

    

     

    
    

     

                  success, frame = cap.read()
     
    
   

    

     

    
    

     
            # print('read a new frame:', success)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

                  params = []
     
    
   

    

     

    
    

     

                  params.append(1)
     
    
   

    

     

    
    

     
            if success:
     
    
   

    

     

    
    

     

                      cv2.imwrite(each_video_save_full_path + each_video_name + "_%d.jpg" % frame_count, frame, params)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

                  frame_count += 1
     
    
   

    

     

    
    

     

              cap.release()
     
    
   

    

     

    
    

     

      np.save('label_dir.npy', label_dir)
     
    
   

    

     

    
    

     

      print(label_dir)
     
    
  
 

文章来源: blog.csdn.net，作者：小小谢先生，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/xiewenrui1996/article/details/108746063