我有NumPy数组x_train,它的形状是(880,):
x_train.shape
(880,)
该数组的每个元素都是具有形状(256,192,3)的图像
x_train[0].shape
(256, 192, 3)
如何重塑x_train数组以获得大小为(880,256,192,3)的数组
如果您试图创建一个numpy
数组,并得到了该形状,这意味着并非所有元素都具有相同的形状。 您可以这样验证:
for ix, item in enumerate(xtrain):
assert item.shape == (256, 192, 3), f'found problematic item at index {ix}'
然后删除有问题的行。