Numpy从np数组中移除一个维度。

11 浏览
0 Comments

Numpy从np数组中移除一个维度。

我有一些要处理的图像,问题是有两种类型的图像,都是106 x 106像素,一些是彩色的,一些是黑白的。

其中一种只有两个(2)维度:

(106,106)

另一种有三个(3)维度:

(106,106,3)

有没有办法我可以去掉最后一个维度?

我尝试了np.delete,但似乎没有起作用。

np.shape(np.delete(Xtrain[0], [2] , 2))
Out[67]: (106, 106, 2)

0