pytorch與numpy中的通道交換問題

pytorch網絡輸入圖像的格式為（C, H, W)，而numpy中的圖像的shape為（H,W,C）

所以一般需要變換通道，將numpy中的shape變換為torch中的shape。

方法如下：

# A是numpy數據類型
A = A.transpose(0,1,2)  # 沒有改變：(h,w,c)
A = A.transpose(2,0,1)   # 轉換為：（c,h,w）

這樣就可以直接輸入到torch的網絡中了。