Question

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

(samples, time, rows, cols, channels)

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

(samples, rows, cols, channels)

如何将timetep信息添加到数据集？框架本身是有序的，但我不知道如何将时间步长数据添加到数据集。

我的当前代码将图像拉成一个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)

在数据集中添加时间步长到视频帧？

1 个答案: