pytorch：讀取的圖像變換通道

由于讀取到的圖像是W*H*C 的格式，但是在輸入網絡的時候要求是C * W * H，是以在輸入網絡之前需要變換通道，通常采用transpose函數實作：

img = img.transpose(2, 0, 1)