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的网络中了。

原文：https://www.cnblogs.com/mengxiangtiankongfenwailan/p/11994242.html