如何重塑包含图像d的数组

Question

我的numpy数组X_train中有4554个图像，数组的形状如下所示。在

print(np.shape(X_train))
(4554,) # TOtal numbe of images
X_train[0].shape
(120, 120, 4) # Each image is 120x120 with 4 channels.

现在我想把数组改成(4554, 120, 120, 4)，这样当我打印时

^{pr2}$

它给我的形状是(4554, 120, 120, 4)，而不是{}。在

我尝试了下面的整形方法，但它给了我错误。在

X_train=X_train.reshape((X_train.shape[0],X_train[0].shape))

错误：TypeError:“tuple”对象不能解释为整数

Answer 1

你在找数字堆栈（）方法。在

如果你有一个3d矩阵列表，你可以制作一个4d矩阵，如下所示：

numpy.stack(your_list_of_training_data, axis=0)

有关说明，请参阅此处的文档：https://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.stack.html