Opencv笔记（三）——视频的获取及保存

一、利用摄像头获取视频

我们经常需要使用摄像头捕获实时图像。OpenCV 为这中应用提供了一个非常简单的接口。让我们使用摄像头来捕获一段视频,并把它转换成灰度视频显示出来。了获取视频,你应该创建一个 VideoCapture 对象。他的参数可以是设备的索引号,或者是一个视频文件。设备索引号就是在指定要使用的摄像头。一般的笔记本电脑都有内置摄像头。所以参数就是 0。你可以通过设置成 1 或者其他的来选择别的摄像头。之后,你就可以一帧一帧的捕获视频了。但是最后,别忘了停止捕获视频。

import numpy as np
import cv2

cap = cv2.VideoCapture(0)
while(True):
    ret, frame = cap.read()
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('opencv',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break;
cap.release()
cv2.destroyAllWindows()

• cv2.VideoCapture(0)利用内置摄像头捕获视频，并返回一个对象。
• cap.read() 返回一个布尔值(True/False)。如果帧读取的是正确的,就是 True。所以最后你可以通过检查他的返回值来查看视频文件是否已经到了结尾。
• cap.isOpened()是用来检查摄像头是否被打开的函数，是就返回True,否则就要用cap.open()函数。
• cv2.waitKey(delay)函数的功能是不断刷新图像，频率时间是delay，单位是毫秒。注意这里的delay不能设置为0，这样会始终停留在第一帧视频。
• cap.get(propId)可以用来获得视频的一些参数信息。propld是0到18之间的数字，每个数字代表视频的属性。

参数	propld
cv2.CAP_PROP_POS_MSEC	0
cv2.CAP_PROP_POS_FRAMES	1
cv2.CAP_PROP_POS_AVI_RATIO	2
cv2.CAP_PROP_FRAME_WIDTH	3
cv2.CAP_PROP_FRAME_HEIGHT	4
cv2.CAP_PROP_FPS	5
cv2.CAP_PROP_FOURCC	6
cv2.CAP_PROP_FRAME_COUNT	7
cv2.CAP_PROP_FORMAT	8
cv2.CAP_PROP_MODE	9
cv2.CAP_PROP_BRIGHTNESS	10
cv2.CAP_PROP_CONTRAST	11
cv2.CAP_PROP_SATURATION	12
cv2.CAP_PROP_HUE	13
cv2.CAP_PROP_GAIN	14
cv2.CAP_PROP_EXPOSURE	15
cv2.CAP_PROP_CONVERT_RGB	16
cv2.CAP_PROP_WHITE_BALANCE	17

cv2.CAP_PROP_RECTIFICATION

18

例如,我可以使用 cap.get(3) 和 cap.get(4) 来查看每一帧的宽和高。默认情况下得到的值是 640X480。但是我可以使用 ret=cap.set(3,320)和 ret=cap.set(4,240) 来把宽和高改成 320X240。

二、从文件中获取视频

与从摄像头中捕获一样,你只需要把设备索引号改成视频文件的名字。在播放每一帧时,使用 cv2.waiKey() 设置适当的持续时间。如果设置的太低视频就会播放的非常快,如果设置的太高就会播放的很慢(你可以使用这种方法控制视频的播放速度)。通常情况下 25 毫秒就可以了。

import numpy as np
import cv2

cap = cv2.VideoCapture(‘／home/wl/wl.avi')
while(cap.isOpened()):
    ret, frame = cap.read()
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('opencv',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break;
cap.release()
cv2.destroyAllWindows()

三、保存视频

在我们捕获视频,并对每一帧都进行加工之后我们想要保存这个视频。对于图片来时很简单只需要使用 cv2.imwrite()。但对于视频来说就要多做点工作。我们要创建一个 VideoWriter 的对象。我们应该确定一个输出文件的名字。接下来指定 FourCC 编码(下面会介绍)。播放频率和帧的大小也都需要确定。最后一个是 isColor 标签。如果是 True,每一帧就是彩色图,否则就是灰度图。

• In Fedora: DIVX, XVID, MJPG, X264, WMV1, WMV2. (XVID ismore preferable. MJPG results in high size video. X264 gives
     very small size video)
• In Windows: DIVX (More to be tested and added)

•In OSX : (I don’t have access to OSX. Can some one fill this?)

FourCC 码以下面的格式传给程序,以 MJPG 为例:cv2.cv.FOURCC('M','J','P','G') 或者 cv2.cv.FOURCC(*'MJPG')。

import numpy as np
import cv2

cap = cv2.VideoCapture(0)
fourcc = cv2.cv.FOURCC(*'XVID')
out = cv2.VideoWriter('/home/wl/output.avi',fourcc, 20.0, (640,480))
while(cap.isOpened()):
    ret, frame = cap.read()
    out.write(frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break;
cap.release()
cv2.destroyAllWindows()