LSTM提供随机噪声图像输出

编程问答 2022-06-07

问题描述

感谢您看这个问题！问题说明如下：

我要做什么？

我正在训练LSTM模型，以根据过去的30帧预测视频中的接下来的10帧。数据集包含968张视频图像。

面临的问题

尽管LSTM收敛了，但是经过训练的模型进行的预测会给出随机的噪声图像，即好像模型尚未学习。

数据准备

  1. 使用cv2.VideoCapture获取视频中的所有帧（968）。
  2. 使用cv2.COLOR_BGR2GRAY将所有帧转换为灰度，然后将np.reshape转换为（128x128）图像，然后除以255以将像素值归一化为0-1。
  3. 结果，准备的数据集的尺寸（968x128x128）的值介于0到1之间。
    4. LSTM L输入X被提取为30个连续帧。 X_dim =（898，30，128x128）
  4. LSTM标签y在X. y_dim（898，10x128x128）之后的10个连续帧中被提取
  5. LSTM模型：2个LSTM堆叠，每个100个单元。
  6. 培训：批次大小= 32，时代= 50，Val_split = 0.2

培训结果

Training history

预期的输出类型

enter image description here

实际预测输出

enter image description here

问题 基于以上设置，我认为模型已经足够收敛而不会过度拟合。但是，我无法解决导致输出不佳的原因。希望在此问题上有任何建议或帮助。

谢谢大家！

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

imageimageimagekeraskeraslstmlstmnoisenoisesequencesequencesequence