当前位置：首页 > article >正文

OpenCV库学习之cv2.VideoWriter(函数)

article 2024/9/21 7:48:09

OpenCV库学习之cv2.VideoWriter(函数)

一、简介

cv2.VideoWriter 是 OpenCV 中用于保存视频文件的函数。它可以将帧写入视频文件中，支持设置视频格式、编码器、帧率和帧大小等参数。

二、语法和参数

语法

cv2.VideoWriter(filename, fourcc, fps, frameSize[, isColor])

参数

filename：保存视频的路径和文件名，字符串类型，例如 'output.mp4'。
fourcc：用于指定视频编码的 4 字符代码，例如 'mp4v'，可以通过 cv2.VideoWriter_fourcc() 函数生成。
fps：视频的帧率，浮点型，表示每秒播放的帧数。
frameSize：视频帧的尺寸，以 (宽度, 高度) 的元组形式给出。
isColor（可选）：布尔值，默认为 True，表示视频是彩色的；如果为 False，则表示视频是灰度的。

返回值

返回一个 cv2.VideoWriter 对象，用于写入视频文件。

三、实例

3.1 创建一个简单的视频文件

此示例展示了如何使用 cv2.VideoWriter 创建一个简单的视频文件，将生成的帧写入文件。

代码

import cv2
import numpy as np

# 创建视频写入对象
output_path = 'output_video.mp4'
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = 20
frame_size = (640, 480)
out = cv2.VideoWriter(output_path, fourcc, fps, frame_size)

# 生成并写入一些帧
for i in range(100):
    frame = np.zeros((480, 640, 3), dtype=np.uint8)
    cv2.putText(frame, f'Frame {i+1}', (100, 240), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 255, 255), 2)
    out.write(frame)

# 释放视频写入对象
out.release()

print("视频保存完成")

输出

输出：
视频保存完成

3.2 从摄像头捕获视频并保存到文件

此示例展示了如何捕获摄像头的视频并使用 cv2.VideoWriter 将其保存到文件中。

代码

import cv2

# 打开摄像头
cap = cv2.VideoCapture(0)

# 检查摄像头是否打开成功
if not cap.isOpened():
    print("无法打开摄像头")
    exit()

# 设置视频写入对象
output_path = 'camera_output.mp4'
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = 20
frame_size = (int(cap.get(3)), int(cap.get(4)))
out = cv2.VideoWriter(output_path, fourcc, fps, frame_size)

# 读取摄像头并保存视频
while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    out.write(frame)
    cv2.imshow('frame', frame)

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

# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()

print("视频保存完成")