torchvision将图片转换成Tensor:transforms.ToTensor()

transforms.ToTensor() (1) transforms.ToTensor() 将numpy的ndarray或PIL.Image读的图片转换成形状为(C,H, W)的Tensor格式,且/255归一化到[0,1.0]之间 (2)通道的具体顺序与cv2读的还是PIL.Image读的图片有关系 cv2:(B,G,R) PIL.Image:(R, G, B) import torch i
相关文章
相关标签/搜索