我有一个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
。
不用说,这根本不起作用。我一整天都在努力解决这个问题。
任何帮助将不胜感激!