制作提取帧的视频时出错

Question

我将视频帧提取到一个名为“images”的文件夹中。让我们考虑一下：

a0.jpg a1.jpg a2.jpg a3.jpg 还有……

现在我想制作这些帧的视频。

我有这段代码可以从提取的视频中创建视频

  import cv2
import os


image_folder = 'images'
video_name = 'video.avi'


images = [img for img in os.listdir(image_folder) if img.endswith(".jpg")]
for i in range(len(images)):
    image_name = 'a' + str(i) + '.jpg'
    frame = cv2.imread(os.path.join(image_folder, images[0]))

    height, width, layers = frame.shape

    video = cv2.VideoWriter(video_name, 0, 1, (width, height))

    video.write(cv2.imread(os.path.join(image_folder, image_name)))
cv2.destroyAllWindows()
video.release()

问题是当我制作视频时，我得到了视频但帧是随机排列的，我怎样才能按顺序排列它们？感谢您的帖子我该如何解决这个问题？

Answer 1

正如您所说，您的框架没有按顺序排列，这意味着当您通过os.listdir 获取它们时，它们没有按顺序排列。你还说他们的名字就像 a0、a1 等等，所以你可以这样做：

images = [img for img in os.listdir(image_folder) if img.endswith(".jpg")]
for i in range(len(images)):
    image_name = 'a' + str(i) + '.jpg'
    video.write(cv2.imread(os.path.join(image_folder, image_name)))

编辑

您将视频编写器置于循环中，因此每次都会重新创建它。这次我附上了完整的代码，应该可以工作。

import cv2
import os

image_folder = 'images'
video_name = 'video.avi'

images = [img for img in os.listdir(image_folder) if img.endswith(".jpg")]

frame = cv2.imread(os.path.join(image_folder, images[0]))
height, width, layers = frame.shape
video = cv2.VideoWriter(video_name, 0, 1, (width, height))

for i in range(len(images)):
    image_name = 'a' + str(i) + '.jpg'
    video.write(cv2.imread(os.path.join(image_folder, image_name)))

cv2.destroyAllWindows()
video.release()