RandomHorizontalFlip

class paddle.vision.transforms. RandomHorizontalFlip ( prob=0.5, keys=None ) [源代码]

基于概率来执行图片的水平翻转。

参数

  • prob (float,可选) - 图片执行水平翻转的概率,取值范围为[0, 1],默认值为 0.5。

  • keys (list[str]|tuple[str],可选) - 与 BaseTransform 定义一致。默认值:None。

形状

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

  • output (PIL.Image|np.ndarray|Paddle.Tensor) - 返回概率执行水平翻转后的图像数据。

返回

计算 RandomHorizontalFlip 的可调用对象。

代码示例

>>> import paddle
>>> fake_img = paddle.to_tensor([[[0, 0, 1], [0, 0, 1], [1, 1, 1]]])
>>> print(fake_img)
Tensor(shape=[1, 3, 3], dtype=int64, place=Place(gpu:0), stop_gradient=True,
       [[[0, 0, 1],
         [0, 0, 1],
         [1, 1, 1]]])
>>> transform = paddle.vision.transforms.RandomHorizontalFlip(prob=1)
>>> result = transform(fake_img)
>>> print(result)
Tensor(shape=[1, 3, 3], dtype=int64, place=Place(gpu:0), stop_gradient=True,
       [[[1, 0, 0],
         [1, 0, 0],
         [1, 1, 1]]])