小编Sar*_*hak的帖子

在numpy中重塑数组

我有一个大小为 numpy 的数组5000x32x32x3。数字 5000 是图像的数量,每个图像都有32x32宽度和高度,并且有3颜色通道。

现在我想以5000x3x32x32保留数据的方式创建一个 numpy 形状数组。我所说的保留数据的意思是:

  1. 结果数组中应该有 5000 个数据点
  2. 3数组的第 2 维 ( ) 正确确定了颜色通道,即所有第 2 维0属于红色通道、第 2 维1属于绿色通道、第 2 维2属于蓝色通道的元素。

简单地重塑 bynp.reshape(data,(5000,3,32,32))是行不通的,因为它不会保留通道,而只是将数据重塑为所需的形状。

python arrays numpy

3
推荐指数
1
解决办法
2027
查看次数

标签 统计

arrays ×1

numpy ×1

python ×1