将 numpy 图像列表转换为 mp4 视频答案

【问题标题】：Converting numpy list of images to mp4 video将 numpy 图像列表转换为 mp4 视频
【发布时间】：2020-04-19 15:28:30
【问题描述】：

我有 1000 个 RGB 图像（1000、96、96、3）的 Numpy 列表。我使用 openCV 从这些图像中创建了一个 mp4 视频。我的道路是棕色的，汽车是红色的，但是在我创建视频后它们变成了蓝色。你能告诉我如何避免这个问题吗？

我的代码：

img_array = []
for img in brown_dataset:
    img_array.append(img)

size = (96,96)

out = cv2.VideoWriter('project_brown.mp4',cv2.VideoWriter_fourcc(*'DIVX'),15, size)

for i in range(len(img_array)):
    out.write(img_array[i])
out.release()

视频之前：

视频后：

【问题讨论】：

OpenCV默认使用BGR格式，可以使用cv2.cvtColor翻转
这能回答你的问题吗？ Change color in RGB images
谢谢，它完全有效

标签： image numpy machine-learning video cv2

【解决方案1】：

如 cmets 中所述，OpenCV 默认使用 BGR 格式，您的输入数据集为 RGB。

这是解决问题的一种方法

img_array = []
for img in brown_dataset:
    img_array.append(img)

size = (96,96)

out = cv2.VideoWriter('project_brown.mp4',cv2.VideoWriter_fourcc(*'DIVX'),15, size)

for i in range(len(img_array)):
    rgb_img = cv2.cvtColor(img_array[i], cv2.COLOR_RGB2BGR)
    out.write(rgb_img)
out.release()

【讨论】：