與NumPy的

Question

交換陣列尺寸當我加載圖像與PIL，並將其轉換成一個NumPy的數組：

image = Image.open("myimage.png") 
pixels = np.asarray(image) 

的數據被存儲爲[X] [Y] [頻道]。即，像素[3,5,0]的值將是（3，5）像素，以及該像素的紅色分量。

但是，我正在使用一個庫，它要求圖像格式爲[channel] [x] [y]。因此，我想知道如何做這種轉換？

我知道NumPy有一個reshape函數，但這實際上並不允許您根據需要「交換」尺寸。

任何幫助？謝謝！

Answer 1

爲了獲得尺寸的順序，你想要的，你可以使用轉置方法如下：

image = Image.open("myimage.png") 
pixels = np.asarray(image).transpose(2,0,1)