OpenCV视频捕获

本文概述

• OpenCV VideoCapture
• 从相机捕获视频
• 从文件播放视频
• 保存视频

OpenCV VideoCapture

OpenCV提供用于与摄像机配合使用的VideoCature()函数。我们可以执行以下任务：

• 读取视频, 显示视频并保存视频。
• 从相机拍摄并显示。

从相机捕获视频

OpenCV允许一个简单的界面来使用摄像机(网络摄像机)捕获实时流。它将视频转换为灰度并显示。

我们需要创建一个VideoCapture对象来捕获视频。它接受设备索引或视频文件的名称。相机指定的数字称为设备索引。我们可以通过传递O或1作为参数来选择摄像机。之后, 我们可以逐帧捕获视频。

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

while(True):
    # Capture image 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。

从文件播放视频

我们可以播放文件中的视频。这类似于通过使用文件名更改摄像机索引来从摄像机捕获。时间必须适合cv2.waitKey()函数, 如果时间过长, 视频将变慢。如果时间太短, 那么视频将非常快。

import numpy as np
import cv2

cap = cv2.VideoCapture('filename')

while(cap.isOpened()):
    ret, frame = cap.read()
#it will open the camera in the grayscale mode
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    cv2.imshow('frame', gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

保存视频

cv2.imwrite()函数用于将视频保存到文件中。首先, 我们需要创建一个VideoWriter对象。然后, 我们应该指定FourCC代码和每秒的帧数(fps)。帧大小应在函数内传递。

FourCC是一个4字节的代码, 用于标识视频编解码器。下面给出了保存视频的示例。

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()

它将视频保存在所需的位置。运行上面的代码, 然后查看输出。