transforms.functional.to_tensor 的作用
作者:互联网
import torchvision.transforms as transforms image = transforms.functional.to_tensor(image)
Convert a
PIL Image
ornumpy.ndarray
to tensor. This function does not support torchscript.
https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.functional.to_tensor
Convert a
PIL Image
ornumpy.ndarray
to tensor. This transform does not support torchscript.Converts a PIL Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0] if the PIL Image belongs to one of the modes (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) or if the numpy.ndarray has dtype = np.uint8
In the other cases, tensors are returned without scaling.
https://pytorch.org/vision/stable/_modules/torchvision/transforms/transforms.html#ToTensor
标签:PIL,tensor,Image,functional,transforms,numpy,ndarray 来源: https://www.cnblogs.com/mrlonely2018/p/14953281.html