Объединить последовательность кадров из массива NumPy по горизонтали

Question

У меня есть массив NumPy с размером: img_array.shape = (20, 10, 56, 56, 3), который соответствует 20 различным последовательностям из 10 кадров с размером (56, 56, 3). Что объединить все 10 кадров каждой из этих последовательностей в одно большее изображение. Поэтому для вывода нового массива NumPy с размером (20, 56, 560, 3), следовательно, для каждой последовательности должно быть изображение (56, 560, 3).

. Как можно объединить эти кадры? Я пытался использовать функцию NumPy reshape, но результат не был правильным, я пытался:

img_array = np.reshape(img_array, (20, 56, 560, 3))

Пока изображение было создано, и я мог сохранить и увидеть результат, это были неправильные строки были смешивать вместе. Как правильно соединить изображения по горизонтали? Может быть, с помощью hstacks, но я не уверен, как именно я могу это сделать.

Answer 1

Вы можете swapaxes и reshape до указанной формы:

a.swapaxes(2,1).reshape(20, 56, 560, 3)

Answer 2

Используйте np.stack, который может складываться вдоль вашего любимого измерения: https://www.geeksforgeeks.org/numpy-stack-in-python/