python tools实现视频的每一帧提取并保存_Python

preface

最近在做 video caption 相关，要处理大量视频。

今天碰到一个问题，就是要将 youtubeclips 数据集中的 avi 格式的视频，将其视频中的每一帧提取出来。之后用 high accuracy optical flow estimation based on a theory for warping 提出的 optical flow（光流），提取运动的光流特征。

method 1

方法 1 是最简单的，用 ffmpeg 工具来完成。

具体的网上有很多这方面的资料，本人只是简单了解了一下如何使用。如下图，有一个名为 ffmpeg_test.avi 的视频：

python tools实现视频的每一帧提取并保存

在当前目录打开终端，输入如下命令：

python" id="highlighter_198477">
			
				?

									$ffmpeg -i ffmpeg_test.avi frames_%03d.jpg -hide_banner

以上我没有指定太多的参数，实际上有很多参数可以指定，如起止的时间，几秒钟取一帧等等。

输入即可获得每一帧。

method 2

下面就是可以用 cv2 模块中的 videocapture、videowriter 来提取了，具体代码如下：

				?

									#! encoding: utf-8

									import os

									import cv2

									import cv

									videos_src_path = '/home/ou-lc/chenxp/downloads/youtube/youtube_select'

									videos_save_path = '/home/ou-lc/chenxp/downloads/youtube/youtube_frames'

									videos = os.listdir(videos_src_path)

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

									for each_video in videos:

									 print each_video

									 # get the name of each video, and make the directory to save frames

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

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

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

									 # get the full path of each video, which will open the video tp extract frames

									 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()

									 print 'read a new frame: ', success

									 params = []

									 params.append(cv.cv_imwrite_pxm_binary)

									 params.append(1)

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

									 frame_count = frame_count + 1

									cap.release()

在最后，我将每一帧保存为 ppm 格式。因为我需要调用之前的 optical flow 论文中的 of 程序，来提取 optical flow image（光流图）。

保存时，根据 opencv 的 doc：opencv 2.4.9 cv2.imwrite，其参数的指定方式如上。一开始在这里跌了好几个跟头，因为不知道如何将参数正确的指定。

reference

http://stackoverflow.com/questions/33311153/python-extracting-and-saving-video-frames
http://stackoverflow.com/questions/12216333/opencv-imread-imwrite-increases-the-size-of-png

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。

原文链接：https://blog.csdn.net/u010167269/article/details/53268686

python tools实现视频的每一帧提取并保存

相关文章

热门资讯