Python 使用Python OpenCV写入mp4视频

在本文中，我们将介绍如何使用Python和OpenCV库来写入mp4视频文件。Python和OpenCV是一对强大的工具，可以实现图像和视频处理的各种功能。

准备工作

在开始之前，我们需要安装Python和OpenCV库。可以通过以下命令来安装它们：

pip install opencv-python

安装完成后，我们可以导入OpenCV库并开始编写代码。

import cv2

创建视频的基本设置

首先，我们需要创建一个VideoWriter对象来写入视频文件。VideoWriter对象需要设置文件名、编解码器、帧率和分辨率等信息。我们可以使用以下代码来创建一个VideoWriter对象：

output_file = "output.mp4"
fourcc = cv2.VideoWriter_fourcc(*"mp4v")
frame_rate = 30.0
resolution = (640, 480)

video_writer = cv2.VideoWriter(output_file, fourcc, frame_rate, resolution)

在上面的代码中，我们指定了输出文件名为output.mp4，使用mp4v编解码器，帧率为30.0，分辨率为640×480。

写入帧

接下来，我们可以开始写入视频的每一帧。我们可以使用OpenCV的cv2.imshow()函数来显示每一帧，然后通过cv2.waitKey()函数来等待用户输入，最后使用cv2.destroyAllWindows()函数来关闭窗口。

下面是一个例子，演示如何写入视频的每一帧：

cap = cv2.VideoCapture(0)
while True:
    ret, frame = cap.read()
    if not ret:
        break

    # 在视频帧上进行一些处理操作
    processed_frame = process_frame(frame)

    # 显示视频帧
    cv2.imshow("Frame", processed_frame)

    # 将处理后的视频帧写入文件
    video_writer.write(processed_frame)

    # 按下q键退出循环
    if cv2.waitKey(1) == ord("q"):
        break

cap.release()
cv2.destroyAllWindows()
video_writer.release()

在上面的例子中，我们通过cv2.VideoCapture(0)函数获取摄像头的视频流。然后，我们使用一个循环来读取每一帧，直到视频结束或用户按下q键。在每一帧上，我们可以进行一些处理操作，比如图像滤波、边缘检测等。然后，我们将处理后的视频帧显示出来，并将其写入到文件中。