如何使用 python、matplotlib 和 opencv 生成视频？

Question

这就是我现在所拥有的。

导入的库：

from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
from matplotlib.figure import Figure
import matplotlib.pyplot as plt
import numpy as np
import cv2
import time

例如（这部分代码来自互联网示例），我画了一个 plot 上面有一个“文本”。

fig = plt.figure(figsize=(6,6.5))
canvas = FigureCanvas(fig)
ax = fig.gca()
ax.text(0.0,0.0,"Test", fontsize=45)
ax.axis('off')

然后，我将它保存到“1.png”。

plt.savefig("1.png")

最后，我创建了一个视频编写器并将 png 写入视频。

size = (600,650)
fps = 2
fourcc = cv2.VideoWriter_fourcc('I', '4', '2', '0')
video = cv2.VideoWriter("1.avi", fourcc, fps, size)
image = cv2.imread('1.png')
video.write(image)
video.release()

这里是问题：我需要保存图片，然后将其读取为 opencv 可以读取的 numpy 数组，有什么方法可以直接将 fig ZA8CFDE6331BD59EB2AC96F8911C4B666py 数组转换为图像对象？

非常感谢你们！

Answer 1

让我们一步一步分析：

1.  size = (600,650)

图像的大小是600, 650吗？

如果我是你，我会初始化为

image = cv2.imread('1.png')
(height, width) = image.shape[:2]
size = (width, height)

或者，如果您希望 output 为(600, 650) ，则调整图像大小：

image = cv2.imread('1.png')
image = cv2.resize(image, (600, 650))

2.  fps = 2

为什么？ 您将每秒帧数设置为 2，为什么不设置为 10？、15？ 资源

3. fourcc = cv2.VideoWriter_fourcc('I', '4', '2', '0')

为什么选择I420 ？ 您要创建.avi文件，然后设置为MJPG 源

如果我是你，我会初始化为：

fourcc = cv2.VideoWriter_fourcc(*'MJPG')

4.  video = cv2.VideoWriter("1.avi", fourcc, fps, size)

您正在使用什么类型的图像？ 颜色，灰色？ 您需要在VideoWriter中初始化为参数

对于彩色图像：

video = cv2.VideoWriter("1.avi", fourcc, fps, size, isColor=True)

对于灰色图像：

video = cv2.VideoWriter("1.avi", fourcc, fps, size, isColor=False)

示例代码：

---

import cv2

img = cv2.imread("1.png")
(h, w) = img.shape[:2]
size = (w, h)
fps = 20
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
video = cv2.VideoWriter("1.avi", fourcc, fps, size, isColor=True)
video.write(img)
video.release()

也知道cv2.imread以BGR方式读取图像。 如果您希望图像为RGB ，则需要将其转换为：

image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)