ToTensor

class torchvision.transforms.ToTensor

将 PIL Image 或 ndarray 转换为 tensor 并相应地缩放值。

此变换不支持 torchscript。

将范围在 [0, 255] 的 PIL Image 或 numpy.ndarray (H x W x C) 转换为形状为 (C x H x W)、范围在 [0.0, 1.0] 的 torch.FloatTensor，前提是 PIL Image 属于以下模式之一 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1)，或者 numpy.ndarray 的 dtype 为 np.uint8。

在其他情况下，返回的 tensors 不进行缩放。

注意

由于输入图像被缩放至 [0.0, 1.0] 范围，因此在对目标图像 mask 进行变换时，不应使用此变换。请参阅参考资料，了解如何实现图像 mask 的变换。