python读取视频流提取视频帧的方法

方法一:通过imageio库和skimage库

  1. 安装环境:

pip install imageio

pip install skimage

这时候会报错Please install the scikit-image package (instead of skimage)

所以按照提示操作即可:

pip install scikit-image

环境安装成功。

2.通过python安装ffmpge(用来解码视频流)

import imageio

imageio.plugins.ffmpeg.download()

这时候会提示:

Imageio: ‘ffmpeg.linux64’ was not found on your computer; downloading it now.Try 1. Download from https://github.com/imageio/imageio-binaries/raw/master/ffmpeg/ffmpeg.linux64 (27.2 MB)Downloading: 8192/28549024 bytes

开始进行下载ffmpeg

注意:下载这个文件很容易断线,所有可以先拿下载工具下载后,再放到下面路径,然后系统就会认为文件已经下载完了。
文件路径: /root/.imageio/ffmpeg/ffmpeg.linux6

如果不断线的话,文件就会存放在上述路径中。

下次执行就可以直接读取视频流了。

  1. 使用imageio库提取视频帧

imageio官方文档地址:http://imageio.readthedocs.io/en/latest/examples.html

用法:

coding:utf-8

import pylab
import imageio
#注释的代码执行一次就好,以后都会默认下载完成
#imageio.plugins.ffmpeg.download()
import skimage
import numpy as np

#视频的绝对路径
filename = ‘video/01_01.wmv’
#可以选择解码工具
vid = imageio.get_reader(filename, ‘ffmpeg’)
for num,im in enumerate(vid):
#image的类型是mageio.core.util.Image可用下面这一注释行转换为arrary
print im.mean()
image = skimage.img_as_float(im).astype(np.float64)
fig = pylab.figure()
fig.suptitle(‘image #{}’.format(num), fontsize=20)
pylab.imshow(im)
pylab.show()

方式二:通过cv2库
在Anaconda Prompt命令窗口输入pip install opencv-python命令安装opencv
用cv2库,用这种方法的好处是返回的就是arrary,不用转换,但方法一可以指定显示某一帧,而这种方法是从头读到尾。
用法:

import cv2

video_full_path = “1.avi”
cap = cv2.VideoCapture(video_full_path)
#print(cap.isOpened())
frame_count = 0
success, frame = cap.read()
while (success):
frame_count = frame_count + 1
# print('Read a new frame: ', success)
params = []
# params.append(cv.CV_IMWRITE_PXM_BINARY)
params.append(1)
#cv2.imwrite(“video” + “_%d.jpg” % frame_count, frame, params)
cv2.imshow(‘image’, frame)
k = cv2.waitKey(50)
# q键退出
if (k & 0xff == ord(‘q’)):
break
success, frame = cap.read()
cap.release()
cv2.destroyAllWindows()


作者:NFMSR
来源:CSDN
原文:https://blog.csdn.net/nfmsr/article/details/78559930
版权声明:本文为博主原创文章,转载请附上博文链接!

个人将第一种方法修改了一下
import pylab
import imageio
#注释的代码执行一次就好,以后都会默认下载完成
#imageio.plugins.ffmpeg.download()
import skimage
import numpy as np

#视频的绝对路径
filename = ‘video/01_01.wmv’
#可以选择解码工具
vid = imageio.get_reader(filename, ‘ffmpeg’)
fig = pylab.figure()

for num, im in enumerate(vid):
# image的类型是mageio.core.util.Image可用下面这一注释行转换为arrary
pylab.pause(0.1)
pylab.imshow(im)
pylab.show()
这样就可以更好的显示视频了。

你可能感兴趣的:(python学习,数据读取)