Transpose

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

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

参数

• order (list|tuple, optional) - 目标的维度顺序. Default: (2, 0, 1)。

• keys (list[str]|tuple[str], optional) - 与 BaseTransform. 默认值: None。

返回

PIL.Image 或 numpy ndarray，更改格式后的图像。

代码示例

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)