torchvision.transforms 数据预处理：ToTensor()

原创

已于 2023-07-06 21:40:35 修改 · 2.4w 阅读

125 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #pytorch

于 2022-11-10 12:33:22 首次发布

本文详细解读了ToTensor()在PyTorch中的作用，包括数据转换、PIL与OpenCV图像读取的比较，以及关键知识点的应用实例。着重介绍了如何处理不同数据类型和格式，使之适用于深度学习模型输入。

该文章已生成可运行项目，

文章目录

ToTensor() 是pytorch中的数据预处理函数，包含在 torchvision.transforms 模块下。一般用于处理图像数据，所以其处理对象是 PIL Image 和 numpy.ndarray 。

1、ToTensor() 函数的作用

必须要声明不能只看函数名，就以为 ToTensor() 只是将图像转为 tensor，其实它的功能不止于此

看一下 ToTensor() 函数的源码：

class ToTensor:
    """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.

    .. note::
        Because the input image is scaled to [0.0, 1.0], this transformation should not be used when
        transforming target image masks. See the `references`_ for implementing the transforms for image masks.

    .. _references: https://github.com/pytorch/vision/tree/main/referen