python - Keras 训练 CNN - 我应该将热图数据转换为图像还是 2D 矩阵

Question

我对训练 Keras CNN 感兴趣，并且我有一些 2D 矩阵形式的数据(例如宽度 x 高度)。我通常使用颜色条来表示或可视化数据，例如热图。

但是，在训练 CNN 和格式化数据输入时，我想知道是否应该将该矩阵保留为 2D 矩阵，还是将其转换为本质上是 3D 矩阵的 RGB 图像？

最佳实践是什么以及人们应该考虑的一些注意事项是什么？

Answer 1

事实上 - 您需要 reshape 单个数据点以具有 3D 形状 - 因为 keras 期望您的数据集具有形状 (数字示例，宽度、高度、 channel )。如果您不想将图像设置为 RGB - 您只需将其保留一个 channel (并将其解释为灰度 channel )。