torch.Tensor 与 numpy.ndarray的相互转化

前言

 在深度学习中，图像的数据类型为torch，其形状（shape）为：(C, H, W)。在opencv中图像的数据类型为ndarray其形状为：（H, W, C）
 如果需要使用opencv显示图像则需要将torch转化为ndarray; 如果需要使用该图像作为深度学习的数据则需要将该图像转化为torch

np.ndarray转为torch.Tensor

np.transpose( xxx,  (2, 0, 1))   # 将 H x W x C 转化为 C x H x W

torch.Tensor转numpy.ndarray

 np.transpose(tensor_cv.numpy(), (1, 2, 0)) # 将C x H x W 转化为 H x W x C 

参考

pytorch学习(五)—图像的加载/读取方式