python opencv 读取指定帧 或将整个视频读取为numpy array

直接上代码:

    假设需要获取某视频的第50帧:

import cv2

cap = cv2.VideoCapture('XXX.avi')  #返回一个capture对象
cap.set(cv2.CAP_PROP_POS_FRAMES,50)  #设置要获取的帧号
a,b=cap.read()  #read方法返回一个布尔值和一个视频帧。若帧读取成功,则返回True
cv2.imshow('b', b)
cv2.waitKey(1000)

    

    假设需要将整个视频读取成一个四阶的numpy array,即它的shape为(帧数,高,宽,通道数3)。首先需要通过capture对象的get方法获取视频的参数信息,再去建立这个array,循环读进数组中。

import numpy as np
import cv2

cap = cv2.VideoCapture('XXX.mp4')
wid = int(cap.get(3))
hei = int(cap.get(4))
framerate = int(cap.get(5))
framenum = int(cap.get(7))

video = np.zeros((framenum,hei,wid,3),dtype='float16')
cnt = 0
while(cap.isOpened()):
    a,b=cap.read()
    cv2.imshow('%d'%cnt, b)
    cv2.waitKey(20)
    b = b.astype('float16')/255
    video[cnt]=b
    print(cnt)
    cnt+=1

    

    capture对象的get方法能够读取到的视频信息如下图所示(从索引0开始):

 

 

评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值