把图片序列合成视频:
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?)