transforms.functional.to_tensor 的作用


import torchvision.transforms as transforms

image = transforms.functional.to_tensor(image)

 

Convert a PIL Image or numpy.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 or numpy.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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM