【python】终于解决了cv2.VideoWriter生成视频后视频文件始终为1KB且无法播放的问题

把图片序列合成视频：

import cv2
import imageio
import os

path = r'D:\dataset\images'
dir_name = os.listdir(path)
for dir in dir_name:
    dir_path = os.path.join(path, dir)
    img = imageio.imread(os.path.join(dir_path, os.listdir(dir_path)[0]))
    vid_writer = cv2.VideoWriter(path + rf'\{dir}.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 24, (img.shape[1], img.shape[0]))
    for i in os.listdir(dir_path):
        vid_writer.write(imageio.imread(os.path.join(dir_path, i)))

结果：

打开显示：

这个程序我之前也用过，可以正常合成视频，但是这次竟然不行，我想可能是图片的问题，所以我比较了之前我使用的图片和这次的图片，发现之前读入的图片格式为(w, h, 3)，而这次的格式为(w, h)，也就是说这次读入的是灰度图片，我猜测视频生成失败的原因可能是cv2.VideoWriter不适用于灰度图片，所以把图片都转换为了BGR形式。

import cv2
import imageio
import os

path = r'D:\dataset\images'
dir_name = os.listdir(path)
for dir in dir_name:
    dir_path = os.path.join(path, dir)
    img = imageio.imread(os.path.join(dir_path, os.listdir(dir_path)[0]))
    vid_writer = cv2.VideoWriter(path + rf'\{dir}.mp4', cv2.VideoWriter_fourcc(*'mp4v'), 24, (img.shape[1], img.shape[0]))
    for i in os.listdir(dir_path):
        # vid_writer.write(imageio.imread(os.path.join(dir_path, i)))  ### 这行改为以下三行
        img = imageio.imread(os.path.join(dir_path, i))
        img2 = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
        vid_writer.write(img2)

现在是正常的了：

补充：
cv2.COLOR_GRAY2RGB将灰度图转变为BGR图，用灰度值Y替换所有B, G, R通道，所以B=Y, G=Y, R=Y。（来源：What does cv2.COLOR_GRAY2RGB do?）