ToTensor¶
- class torchvision.transforms.ToTensor[源代码]¶
将 PIL Image 或 ndarray 转换为张量并相应地缩放值。
此转换不支持 torchscript。
将范围在 [0, 255] 内的 PIL Image 或 numpy.ndarray (H x W x C) 转换为范围在 [0.0, 1.0] 内的形状为 (C x H x W) 的 torch.FloatTensor,如果 PIL Image 属于以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) 或如果 numpy.ndarray 的 dtype = np.uint8
在其他情况下,张量将不进行缩放返回。
注意
由于输入图像被缩放至 [0.0, 1.0],因此在转换目标图像掩码时不应使用此转换。有关实现图像掩码转换的信息,请参阅 参考。