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

2024-03-29 08:40:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个numpy arraybatch初始化如下:

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

它应该是一个由50个不同的,60帧/秒的视频组成的阵列,尺寸为1920x1080,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。你知道吗

不用说,这根本不管用。我花了一天的大部分时间想弄明白这一点。你知道吗

任何帮助都将不胜感激!你知道吗


Tags: numpy视频尺寸npbatch时间zerosbatches