CLASStorchvision.transforms.ToTensor[SOURCE]
Convert a PIL Image or numpy.ndarray to tensor.
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.
本文详细介绍了torchvision.transforms.ToTensor的功能,它能将PIL Image或numpy.ndarray转换为torch.FloatTensor,并调整其形状从(HxWxC)到(CxHxW),适用于多种图像模式,如RGB、灰度等。在特定情况下,此转换会进行缩放处理。
3013

被折叠的 条评论
为什么被折叠?



