OpenCV Python - 从摄像头捕获视频

通过使用 OpenCV 库中的 **VideoCapture()** 函数，可以非常轻松地在 OpenCV 窗口中从摄像头捕获实时流。

此函数需要一个设备索引作为参数。您的计算机可能连接了多个摄像头。它们由从 0 开始的索引枚举，内置网络摄像头为 0。该函数返回一个 VideoCapture 对象。

cam = cv.VideoCapture(0)

打开摄像头后，我们可以借助 **read()** 函数读取其连续帧。

ret,frame = cam.read()

read() 函数读取下一个可用的帧并返回一个值（True/False）。此帧现在使用 **cvtColor()** 函数渲染为所需的色彩空间，并在 OpenCV 窗口中显示。

img = cv.cvtColor(frame, cv.COLOR_BGR2RGB)
# Display the resulting frame
cv.imshow('frame', img)

要将当前帧捕获到图像文件，可以使用 **imwrite()** 函数。

cv2.imwrite(“capture.png”, img)

要将来自摄像头的实时流保存到视频文件，OpenCV 提供了 VideoWriter() 函数。

cv.VideoWriter( filename, fourcc, fps, frameSize)

fourcc 参数是视频编解码器的标准化代码。OpenCV 支持各种编解码器，例如 DIVX、XVID、MJPG、X264 等。fps 和 framesize 参数取决于视频捕获设备。

VideoWriter() 函数返回一个 VideoWrite 流对象，捕获的帧将依次写入循环中。最后，释放帧和 VideoWriter 对象以完成视频的创建。

示例

以下示例从内置网络摄像头读取实时视频流并将其保存到 output.avi 文件。

import cv2 as cv
cam = cv.VideoCapture(0)
cc = cv.VideoWriter_fourcc(*'XVID')
file = cv.VideoWriter('output.avi', cc, 15.0, (640, 480))
if not cam.isOpened():
   print("error opening camera")
   exit()
while True:
   # Capture frame-by-frame
   ret, frame = cam.read()
   # if frame is read correctly ret is True
   if not ret:
      print("error in retrieving frame")
      break
   img = cv.cvtColor(frame, cv.COLOR_BGR2RGB)
   cv.imshow('frame', img)
   file.write(img)

   
   if cv.waitKey(1) == ord('q'):
      break

cam.release()
file.release()
cv.destroyAllWindows()