python实现批量视频分帧、保存视频帧


						本篇博客介绍利用python脚本实现视频分帧，并将每一帧保存到本地。主要基于opencv包来实现，在运行代码前确保opencv包已正确安装。下面是主要代码：





import os 

import cv2

 

videos_src_path = '/home/shao/violence_detection_code/Movies_Dataset/fights'

videos_save_path = '/home/shao/violence_detection_code/Movies_Dataset/fights_pic'

 

videos = os.listdir(videos_src_path)

videos.sort(key=lambda x:int(x[5:-4]))

 

i = 1

 

for each_video in videos:

 if not os.path.exists(videos_save_path + '/' + str(i)):

 os.mkdir(videos_save_path + '/' + str(i))

 each_video_save_full_path = os.path.join(videos_save_path,str(i))+'/'

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

 cap = cv2.VideoCapture(each_video_full_path)

 frame_count = 1

 success = True

 

 while(success):

 success,frame = cap.read()

 if success==True:

 cv2.imwrite(each_video_save_full_path + "frame%d.jpg" % frame_count,

 frame)

 frame_count = frame_count + 1

 i = i + 1

 

 cap.release()



代码注释：

1.第1,2行完成导入几个需要的包；

2.第4,5行分别是待分帧的原始视频路径、分帧后需要保存视频帧的文件路径；这里需要注意的是，我写的是文件夹，也就是说在/home/shao/violence_detection_code/Movies_Dataset/fights目录下有多个视频XXX.avi,XXX.avi等形式；如果只需要处理单个视频的话，可以稍微改一下即可。

3.第7行表示将所有视频的文件名存到videos中，第8行是可选项，表示对fights文件夹下的视频进行排序，x[5:-4]表示按照文件名第5个字符到倒数第4个字符之间的符号排序，因为我的视频是newfi1.avi,newfi2.avi……的格式，我想实现的是按照数字1,2,3的顺序提取视频帧；

4.之后的代码就是循环处理每个视频，将每个视频的视频帧保存到文件夹中。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持中文源码网。