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
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】博客园2025新款「AI繁忙」系列T恤上架,前往周边小店选购
【推荐】凌霞软件回馈社区,携手博客园推出1Panel与Halo联合会员
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步