OpenCV 视频捕获|极客笔记

OpenCV 视频捕获

OpenCV提供了 VideoCapture() 函数，用于与摄像头一起工作。我们可以执行以下任务：

读取视频，显示视频和保存视频。
从摄像头捕获并显示。

从摄像头捕获视频

OpenCV允许使用摄像头（网络摄像头）捕获实时流的简单界面。它会将视频转换为灰度并显示出来。

我们需要创建一个 VideoCapture 对象来捕获视频。它可以接受设备索引或视频文件的名称。指定给摄像头的数字称为设备索引。我们可以通过传递0或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()

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