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:breakout.write(frame)cv2.imshow('frame', frame)if cv2.waitKey(1) & 0xFF == ord('q'):break# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()print("视频保存完成")
- 输出
输出：
视频保存完成
四、注意事项
- 视频编码格式需要与文件扩展名匹配，例如
.mp4使用'mp4v'编码器，.avi通常使用'XVID'。 - 帧的大小必须与
VideoWriter对象的frameSize参数一致，否则可能会导致写入失败。 - 在处理实时视频流时，确保正确释放资源（摄像头和视频写入对象）以避免文件损坏或内存泄漏。