cos*_*sst 4 python arrays numpy image-processing tensorflow
我将 RGB 图像(32 x 32 x 3)保存为 3D numpy 数组,我将其用作神经网络的输入(使用 tensorflow)。为了将它们用作输入,我使用 reshape(1,-1) 将它们重塑为一维 np 数组 (1 x 3072)。当我完成训练我的网络时,我想重新调整输出,但使用 reshape(32,32,3) 似乎没有提供所需的结果。
这是正确的方法吗?我如何确定每个数据都会回到正确的位置?
小智 7
如果您要创建一维数组,请使用 .reshape(-1),这将创建数组的线性版本。如果您使用 .reshape(32,32,3),这将创建一个由 32 个 32×3 数组组成的数组,这是所描述的原始格式。使用“-1”创建一个与组合嵌套数组中元素数量相同大小的线性数组。