向数据集中的视频帧添加时间步长？

Question

我目前正在开发一个ConvLSTM编码器-解码器网络，并尝试为它准备一个数据集。根据Tensorflow的文档，ConvLSTM2D层的建议输入数据格式为：

(samples, time, rows, cols, channels)

原始数据集是一系列图像帧，我已经能够用以下格式准备数据集：

(samples, rows, cols, channels)

如何将timestep信息添加到数据集？帧本身是有序的，但我不知道如何将timestep数据添加到数据集中。你知道吗

我当前将图像拉入numpy数组的代码如下：

import glob
import numpy as np
import imageio

original_data = []

for img_path in sorted(glob.glob("../Datasets/Test_Dataset/*.png")):
    img = imageio.imread(img_path)
    original_data.append(img)    

original_data = np.array(original_data)

Answer 1

original_data = original_data.reshape(numberOfVideos, frames, rows, cols, channels)