PIL讀取圖片的形狀及變形

代碼：
from PIL import Image

img = Image.open(“圖片路徑”).convert(‘RGB’)

此時圖片的形狀是hwc，一般來說，訓練網絡要的形狀是whc，這裏可以用numpy變形
代碼如下
img=np.array(img)
img=img.transpose((1,0,2)) #hwc–>whc