Python:重塑一个numpy数组的3D图像和标签

时间:2017-10-09 18:25:30

标签: python numpy machine-learning tensorflow

我正在尝试将(1397,2)的numpy数组重塑为(1397,50,50,20,1)。数组有2列

each row  in column1 is of length: 50*50*20  [3D image]
each row in column2 is of  lenght : 2 [labels : [0,1],[1,0]].

基本上尝试解压缩第一列3d图像以馈入CNN中的输入层。

到目前为止我的代码

np.array([i[0] for i in data]).reshape(-1,50,50,20,1)

给我一​​个错误:

"ValueError: cannot reshape array of size 1397 into shape (50,50,20,1)"

非常感谢任何帮助

干杯 罗宾

1 个答案:

答案 0 :(得分:0)

好的,因为事实证明第1列中的所有行的形状都不一样。一个是17 * 50 * 50而不是20 * 50 * 50。现在好了:)

感谢迈克尔!