python opencv 入门 加载视频 (2)

来源:互联网 发布:米哈伊洛维奇 知乎 编辑:程序博客网 时间:2024/05/16 14:30

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

目标:

  1. 读取视频、显示视频、保存视频
  2. 使用摄像头拍摄并显示
  3. 学习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()

得到的视频是掉转过来的

0 0
原创粉丝点击