OpenCV-Python:视频

通过摄像头捕获视频

我们经常会要用摄像头来捕获实时视频流。OpenCV提供了一个十分简单的接口来做这个。让我们捕获一段摄像头里的视频，把它转换到灰度视频然后显示它。

要捕获一段视频，你需要创建一个VideoCapture对象。它的参数可以是设备索引或者是视频文件名。设备索引是指定哪个摄像头。一般来说只有一个摄像头。所以我们只需要穿入0（或者-1）。你可以传1来选第二个摄像头，以此类推。这之后，你可以一帧帧的捕获，但是最后，不要忘记释放捕获。

import numpy as np
import cv2
cap = 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 capture
cap.release()
cv2.destroyAllWindows()

cap.read()返回一个布尔值（True/False）。如果帧读入正确，会返回True。所以你可以通过检查这个返回值来判断视频是否结束。

有时候，cap可能没有初始化捕获。在这种情况下，这个代码会报错。你可以通过方法cap.isOpened()来检查它是否初始化了。如果是True，那就是初始化了。否则通过cap.open()打开。

你也可以使用cap.get(propId)来访问这个视频的一些属性，propId是从0到18 的一个数。每个数字代表了视频的一个属性。有些值通过cap.set(propId, value)可以修改。

帧的长宽设置
比如，我可以通过cap.get(3)和cap.get(4)来获得这一帧的宽度和高度。它默认会返回640x480，但是我想把值修改成320x240.只需要用ret = cap.set(3, 320)和ret = cap.set(4, 240)

注意：

如果你得到错误，确保摄像头是在正常工作的。
从文件播放视频

和从摄像头捕获一样，值需要把摄像头索引号换成视频文件名。当显示帧的时候，给cv2.waitKey()使用合适的时间。如果太少，视频会太快，如果太高，视频会很慢（慢速播放）。正常情况下25毫秒是ok的。

import numpy as np
import cv2
cap = 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'):
        break

cap.release()
cv2.destroyAllWindows()

注意：
确认ffmpeg或者gstreamer安装正确，有时候，处理视频是很头疼的事情

保存一个视频
我们捕获的视频，一帧帧的处理，现在我们想保存视频，对于图片，十分简单，值需要cv2.imwrite().但视频需要多一点工作了。

我们这次创建一个VideoWriter对象。我们应该指定输出文件名（比如：output.avi），然后我们应该指定FourCC代码（下段详细介绍）。还有每秒帧数（fps）和帧的大小。最后是isColor标志。如果是True，编码器需要彩色帧。否则就是灰度帧。

FourCC是一个4为的代码，用来指定视频编码。可用的视频编码列表可以在fourcc.org找到。这个是跨平台的。下面这些编码都能用：

·In Fedora: DIVX, XVID, MJPG, X264, WMV1, WMV2. (XVID 更推荐. MJPG results in high size video. X264 gives very small size video)

In Windows: DIVX (More to be tested and added)

FourCC代码通过 cv2.VideoWriter_fourcc(‘M’, ‘J’, ‘P’, ‘G’)传入，或者cv2.VideoWriter_fourcc(*’MJPG’)

下面的代码从摄像头捕获视频，每一帧都垂直翻转，然后保存

import numpy as np
import cv2
cap = cv2.VideoCapture(0)

# Define the codec and create VideoWriter object
fourcc = 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 finished
cap.release()
out.release()
cv2.destroyAllWindows()

猜你喜欢