Изменение способа чтения изображений программой matplotlib

Question

Я хочу создать CNN, и я использую matplotlib для чтения изображений (пытался использовать cv2, но не очень хорошо работал с отображением изображений должным образом, и PIL у меня почему-то не работает), в любом случае, я могу читать изображение, но оно читается как Image [высота] [ширина] [канал]. Как я могу изменить его так, чтобы он читался так: Изображение [канал] [высота] [ширина], чтобы мне было проще свернуть его позже? Конечно, говоря, что я имел в виду не запускать вложенные циклы, чтобы получить каждое значение из каждого канала и поместить их в матрицу, я спрашиваю, есть ли лучший способ.

Я ценю всю помощь:)

Answer 1

Вы сможете добраться туда, куда хотите, используя numpy s moveaxis()

img = np.zeros((5,6,3))
img.shape
>> (5, 6, 3)
img2 = np.moveaxis(img, -1, 0)
img2.shape
>> (3, 5, 6)