HWC是指图片的高、宽、通道;
其中C包含了RGB
cv2:BGR(numpy.ndarray)
PIL Image.open:RGB;HWC;[0,255] (numpy.ndarray)
pytorch:RGB;CHW;[0,1] (torch.Tensor)
所以需要用 .ToTensor进行转换——把HWC [0,255] 变为CHW [0,1]
PS: .ToTensor处理的是PIL图
综上,一般流程是:
PIL读入图片——.ToTensor——愉快的去用pytorch吧
要深度学习:numpy.ndarray转为torch.Tensor
//.ToTensor
torch.tensor(xxx)
//没必要吧: np.transpose( xxx, (2, 0, 1)) # 将 H x W x C 转化为 C x H x W
要显示图像:torch.Tensor转numpy.ndarray
np.transpose(xxx.numpy(), (1, 2, 0)) # 将C x H x W 转化为 H x W x C