【问题标题】:TensorFlow: How to configure the parameters of LSTMTensorFlow:如何配置 LSTM 的参数
【发布时间】:2022-07-28 07:41:53
【问题描述】:

我来自 CNN 模型。 我要学习LSTM。 就我而言,我将对 PNG 图像进行分类。 我对 CNN 的了解是:

  • CNN 的输入是图像的形状。例如(宽度、高度、通道)。在我的情况下是 (128, 128, 3)

但是我如何为此类数据构建 LSTM 模型。 我对 LSTM 的了解是:

  • “unit”参数代表隐藏层中有多少音符。

如何设置 RNN 模型的 input_shape?

【问题讨论】:

    标签: python tensorflow keras lstm


    【解决方案1】:

    取决于您要为网络提供什么。如果你在做视频分类,你的数据格式是(frames, width, height, channels),如果你用RNN做图像分类,那么你有(width, height, channels)

    此时,您应该决定每次要为网络提供什么。

    在第一种情况下,ConvLSTM2D 可能是最好的,因为结合了这两个功能

    在第一种情况下,您的“时间”轴可能是像素,因此您可能希望将输入重塑为 (width x height, channels, 1),并且在每一步中,您都为网络提供“一个像素”(带有相对通道)

    单位是隐藏的(在 TF 中也是输出)大小,它是应该在时间上保存信息的“矩阵”

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2016-10-01
      • 1970-01-01
      • 1970-01-01
      • 2016-06-22
      • 1970-01-01
      • 2017-12-29
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多