从图像和相应的注释文件中录制视频

Question

我将图像以另一种格式存储在一个文件夹中。在一个单独的文件夹中，我有包含应该在图像上绘制的矩形坐标的文本文件。我使用read_file 函数从文件中读取 NumPy 数组。

从下面给出的脚本中，我可以成功地在图像上一个接一个地绘制坐标。但这只是一个接一个地生成图像，而我想要上述图像的视频，即由下面给出的脚本生成的图像。

执行上述操作的代码 sn-p：

for idx, val in enumerate(img_list):
    image = read_file(img_list[idx])
    with open(annot_list[idx], 'r') as f:
        lines = [idx.rstrip('\n') for idx in f.readlines()]
        annots = [list(map(int, idx.split(','))) for idx in lines]
    plt.figure()
    plt.imshow(image, cmap='gray')
    for jdx, annot in enumerate(annots):
        x1 = annots[jdx][0]
        y1 = annots[jdx][1]
        w = annots[jdx][2] - x1
        h = annots[jdx][3] - y1
        rect = patches.Rectangle((x1, y1), w, h, linewidth=3, edgecolor='r', facecolor='none')
        plt.gca().add_patch(rect)
    plt.show()

这里，img_list和annot_list变量包含图像文件的名称和各自注释文件的名称。

这会一个接一个地生成一组图像。如何生成这些帧的视频？我应该调整什么？我看过这个thread，但我似乎无法弄清楚我需要更改什么才能生成视频。我不一定要保存视频，也不想保存帧列表中的帧，因为我有大约 5000 张图像。

谢谢你，我很感激我能得到任何帮助。

Answer 1

首先，在循环之前定义一个视频编写器。在循环中，显示图像后，定义一个图形和一个随机图，以便将图像保存到一个数组中。最后，将每张图片写入视频写入器。

out = cv2.VideoWriter('output.avi', -1, 20.0, (640, 480)) # Change the 640 and 480 to the dimensions of your images

for idx, val in enumerate(img_list):
    image = read_file(img_list[idx])
    with open(annot_list[idx], 'r') as f:
        lines = [idx.rstrip('\n') for idx in f.readlines()]
        annots = [list(map(int, idx.split(','))) for idx in lines]
    plt.figure()
    plt.imshow(image, cmap='gray')
    for jdx, annot in enumerate(annots):
        x1 = annots[jdx][0]
        y1 = annots[jdx][1]
        w = annots[jdx][2] - x1
        h = annots[jdx][3] - y1
        rect = patches.Rectangle((x1, y1), w, h, linewidth=3, edgecolor='r', facecolor='none')
        plt.gca().add_patch(rect)
    plt.show()

    fig = plt.figure() # Define figure
    fig.add_subplot(111) # Random plot
    fig.canvas.draw()
    img = np.fromstring(fig.canvas.tostring_rgb(), dtype='uint8', sep='') # Get image array
    out.write(img) # Write to video writer

out.release() # End the video writer

---

如果这部分不起作用：

    fig = plt.figure() # Define figure
    fig.add_subplot(111) # Random plot
    fig.canvas.draw()
    img = np.fromstring(fig.canvas.tostring_rgb(), dtype='uint8', sep='') # Get image array
    out.write(img) # Write to video writer

你可以试试：

    plt.savefig("temp.png")
    img = cv2.imread("temp.png")
    out.write(img) # Write to video writer