paddle.vision.transforms.to_tensor的功能

       paddle.vision.transforms.to_tensor与paddle.to_tensor两者的作用完全不同,paddle.to_tensor只是将其他数据类型转化为tensor类型,便于构建计算图。

        而paddle.vision.transforms.to_tensor是将 PIL.Image 或 numpy.ndarray 转换成 paddle.Tensor

将形状为 (H x W x C)的输入数据 PIL.Image 或 numpy.ndarray 转换为 (C x H x W)。 如果想保持形状不变,可以将参数 data_format 设置为 'HWC'

同时,如果输入的 PIL.Image 的 mode 是 (L, LA, P, I, F, RGB, YCbCr, RGBA, CMYK, 1) 其中一种,或者输入的 numpy.ndarray 数据类型是 'uint8',那个会将输入数据从(0-255)的范围缩放到 (0-1)的范围。其他的情况,则保持输入不变。

你可能感兴趣的:(python,paddle,paddle,python)