本篇博客介绍利用python脚本实现视频分帧,并将每一帧保存到本地。主要基于opencv包来实现,在运行代码前确保opencv包已正确安装。下面是主要代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
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.之后的代码就是循环处理每个视频,将每个视频的视频帧保存到文件夹中。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/Jaster_wisdom/article/details/88635279