matplotlib在视频中画线

为了在视频中使用matplotlib画线，我们需要安装matplotlib和OpenCV库。以下是一个示例代码，它创建了一个红色线段并将其保存为.avi文件：

import cv2
import numpy as np
import matplotlib.pyplot as plt

# 设置视频参数
width, height = 640, 480
fps = 30
fourcc = cv2.VideoWriter_fourcc(*"MJPG")
out = cv2.VideoWriter("output.avi", fourcc, fps, (width, height))

# 创建一条红色线段
x = np.arange(0, width)
y = np.ones(width) * height / 2
line = np.array([x, y]).T.astype(int)

# 逐帧绘制线段
for i in range(100):
    img = np.zeros((height, width, 3), dtype=np.uint8)
    img[:, :] = [255, 255, 255]
    cv2.polylines(img, [line], False, (0, 0, 255), thickness=2)
    # 将matplotlib图形转换为OpenCV图像
    fig, ax = plt.subplots(figsize=(6.4, 4.8), dpi=100)
    ax.imshow(img)
    ax.set_axis_off()
    fig.canvas.draw()
    img = np.array(fig.canvas.renderer.buffer_rgba())
    img = cv2.cvtColor(img, cv2.COLOR_RGBA2BGR)
    # 写入视频
    out.write(img)

# 释放资源
out.release()
plt.close()

该代码创建一个640x480像素的.avi文件，其中包含一条红色线段，线段在100帧中逐渐向右移动。在每一帧中，我们使用matplotlib在OpenCV图像上绘制线段，并将其写入视频文件。最后，我们释放资源并关闭matplotlib图形。