如何更改神经网络图像数据形状的维度顺序？

Posted 2023-02-16

【中文标题】如何更改神经网络图像数据形状的维度顺序？

【英文标题】：How to change the order of dimensions of images data's shape for a neural network?

【发布时间】：2022-01-02 12:08:45

【问题描述】：

我有一个灰度图像数据集，我想将其输入神经网络。

x_train_grey.shape 是 (32, 32, 73257) 所以我知道它是 (dimension_x, dimension_y, batch_size)。因为图像是灰度的，所以只有一个“深度”维度。

但是，要将这些数据提供给神经网络，它需要具有以下形状：(batch_size, dimension_x, dimension_y)。以batch_szie 开头。 如何将其重塑为这种格式，以便 batch_szie 出现在 x、y 图像尺寸之前？

一旦完成，我希望能够将其传递到神经网络（第一层是Flatten()），如下所示： Flatten(input_shape=(32, 32, 1)),.

干杯！

【问题讨论】：

你可以查看numpy中的reshape方法

【参考方案1】：

解决了！通过将正确的形状传递给np.reshape()。

在进入深度学习之前，我真的应该更好地了解numpy。