【发布时间】:2020-02-18 14:49:59
【问题描述】:
我正在尝试在 keras 中使用 CONVLSTM2D,但我不知道如何准备和输入训练数据。我不知道输入数据是一张带有要学习的帧序列的图像还是一个图像单元格(如matlab)。
【问题讨论】:
标签: tensorflow keras conv-neural-network lstm
我正在尝试在 keras 中使用 CONVLSTM2D,但我不知道如何准备和输入训练数据。我不知道输入数据是一张带有要学习的帧序列的图像还是一个图像单元格(如matlab)。
【问题讨论】:
标签: tensorflow keras conv-neural-network lstm
既然您提到了帧,我假设您想在视频上训练模型。对于视频,您可以将视频转换为单个 numpy 文件并按照示例代码 sn-p 中所述训练模型。
>>> np.load('dataset/scene_0.npy').shape()
(1000, 3, 400, 400)
>>> np.load('dataset/category_0.npy').shape()
(6, 1, 1000, 1)
然后您需要使用生成器,该生成器使用 .fit_generator() 逐个样本馈送模型样本。
您可以找到有关此主题的详细信息here
【讨论】: