首页 > 编程 > Python > 正文

python opencv 入门 加载视频 (2)

2019-11-06 06:52:13
字体:
来源:转载
供稿:网友

内容来自OpenCV-Python Tutorials 自己翻译整理

目标:

读取视频、显示视频、保存视频使用摄像头拍摄并显示学习cv2.VideoCapture(), cv2.VideoWriter()函数

使用摄像头拍摄视频: (我的用的是笔记本电脑上面的摄像头) OpenCV 提供了一个非常简单的接口。我们把视频转换成黑白格式,然后显示出来。

首先要创建一个VideoCapture对象。参数可以是设备标示(device index)或是一个视频文件的名称。设备标示是指定使用哪个摄像头的编号。通常一个摄像头会连接上。所以只需要传递参数0或者-1。可以选择第二个摄像头通过传递参数1,以此类推。之后就能通过摄像头获得每一帧的数据,在最后释放摄像头资源。

import numpy as npimport cv2cap = cv2.VideoCapture(0)while(True): # Capture frame-by-frame ret, frame = cap.read() # Our Operations on the frame come here gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # Display the resulting frame cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): break# When everything done, release the capturecap.release()cv2.destroyAllWindows()

cap.read() 函数返回一个布尔类型。如果读的视频帧是正确的,返回True,所以可以通过返回值检查。

有时,cap可能没有初始化摄像头,可以检查是否初始化或者使用cap.isOpened() 函数,如果返回True则继续,否则使用cap.open()

可以使用cap.get(PRoid)函数来设置视频的属性,proid是0到18的数字每个数组表示一个视频的属性,具体见Property Identifier 其中一些属性可以是用cap.set(proid,value)方法修改

例如,使用cap.get(3)和cap.get(4)获得每一帧的宽度和高度,获得的结果是640和480,可以把它修改成320×240 ret = cap.set(3,320) ret = cap.set(4,240)

#!/usr/bin/env python# coding=utf-8import cv2import numpy as npfrom matplotlib import pyplot as pltcap = cv2.VideoCapture(0)ret = cap.set(3,320)ret = cap.set(4,240)while(True): ret,frame = cap.read() gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY) cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): breakcap.release()cv2.destroyAllWindows()

(不要把set写到循环里,否则会很卡顿)

播放视频文件: 只需要把VideoCapture里面的摄像头标示改成文件名即可,如果视频很小,播放的太快,可以使用参数cv2.waitKey()函数,设置成25毫秒效果不错

import numpy as npimport cv2cap = cv2.VideoCapture('vtest.avi')while(cap.isOpened()): ret, frame = cap.read() gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): breakcap.release()cv2.destroyAllWindows()

存储视频: 要把摄像机拍摄到的视频按帧存起来可以是用cv2.imwrite()函数

首先建立一个VideoWriter对象,然后指定要保存的文件名,例如output.avi。接下来使用FourCC,接下来传递每秒的帧数和帧的大小参数,最后是isColor标记,如果该标记为真,将得到彩色的图片。

FourCC 是用于指定视频编解码器 4 字节代码。代码可以在fourcc.org上找到,该代码是平台相关的。(本人机器为Ubuntu 14.04)

import numpy as npimport cv2cap = cv2.VideoCapture(0)# Define the codec and create VideoWriter objectfourcc = cv2.VideoWriter_fourcc(*'XVID')out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))while(cap.isOpened()): ret, frame = cap.read() if ret==True: frame = cv2.flip(frame,0) # write the flipped frame out.write(frame) cv2.imshow('frame',frame) if cv2.waitKey(1) & 0xFF == ord('q'): break else: break# Release everything if job is finishedcap.release()out.release()cv2.destroyAllWindows()

得到的视频是掉转过来的


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表