如何将数据分配给多维numpy数组?

时间:2019-06-20 15:24:09

标签: python numpy numpy-ndarray matrix-indexing

我有一个numpy array batch初始化如下:

batch = np.zeros((50, 60, 1920, 1080, 3))

应该是由50个分辨率为1920x1080的60FPS视频组成的阵列,而3则代表三个通道-红色,绿色,蓝色。每个视频正好是1秒。

我遍历我的视频文件夹中的所有视频,并对每个视频的每一帧进行图像处理。然后,我将转换后的视频写入batch数组中。如何正确索引批处理数组以保存符合batch数组尺寸的每个视频?

到目前为止,我已经尝试了以下方法:

batch[:batches_produced, :idx, :] = frame[:]

其中batches_produced是当前批次项目的索引,idx是当前框架的索引,frame是维度(1920x1080x3)的实际框架。

当我

print(batch_data[1,2,:,:,:].shape),它会抛出

IndexError: index 1 is out of bounds for axis 0 with size 1

不用说,这根本不起作用。我一整天都在努力解决这个问题。

任何帮助将不胜感激!

0 个答案:

没有答案