ToTensor

class torchvision.transforms.ToTensor

将 PIL 图像或 ndarray 转换为张量并相应地缩放值。

此转换不支持 torchscript。

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

在其他情况下，张量不进行缩放直接返回。

注意

由于输入图像被缩放到 [0.0, 1.0]，因此在转换目标图像掩码时，不应使用此变换。有关实现图像掩码变换的方法，请参阅参考资料。