Transpose¶

class paddle.vision.transforms. Transpose ( order=(2, 0, 1), keys=None ) [源代码] ¶

将输入的图像数据更改为目标格式。例如，大多数数据预处理是使用 HWC 格式的图片，而神经网络可能使用 CHW 模式输入 Tensor。输出的图片是 numpy.ndarray 的实例。

参数¶

order (list|tuple，可选) - 目标的维度顺序。Default: (2, 0, 1)。

keys (list[str]|tuple[str]，可选) - 与 BaseTransform 一致。默认值：None。

形状¶

img (PIL.Image|np.ndarray|Paddle.Tensor) - 输入的图像数据，数据格式为'HWC'。

output (np.ndarray|Paddle.Tensor) - 返回更改格式后的数组或 Tensor。如果输入是 PIL.Image ，输出将会自动转换为 np.ndarray 。

返回¶

计算 Transpose 的可调用对象。

代码示例¶

          import numpy as np
from PIL import Image
from paddle.vision.transforms import Transpose

transform = Transpose()

fake_img = Image.fromarray((np.random.rand(300, 320, 3) * 255.).astype(np.uint8))

fake_img = transform(fake_img)
print(fake_img.shape)