У меня numpy array batch инициализируется следующим образом:
batch = np.zeros((50, 60, 1920, 1080, 3))
Предполагается, что это будет массив из 50 различных видео 60FPS размером 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.
Само собой разумеется, это не работает вообще. Я потратил большую часть своего дня, пытаясь понять это.
Любая помощь будет принята с благодарностью!