前言

 在深度学习中，图像的数据类型为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学习(五)—图像的加载/读取方式