RandomRotation

class paddle.vision.transforms. RandomRotation ( degrees, interpolation='nearest', expand=False, center=None, fill=0, keys=None ) [源代码]

依据 degrees 参数指定的角度范围，按照均匀分布随机产生一个角度对图像进行旋转。

参数

  • degrees (sequence|float|int) - 旋转的角度度数范围。如果度数是数字而不是像（min，max）这样的序列，则会根据 degrees 参数值生成度数范围（-degrees，+degrees）。

  • interpolation (str，可选) - 插值的方法。如果这个参数没有设定或者输入图像为单通道，则该参数会根据使用的后端，被设置为 PIL.Image.NEAREST 或者 cv2.INTER_NEAREST。当使用 pil 作为后端时，支持的插值方法如下：- "nearest": Image.NEAREST,- "bilinear": Image.BILINEAR,- "bicubic": Image.BICUBIC 当使用 cv2 作为后端时，支持的插值方法如下：- "nearest": cv2.INTER_NEAREST,- "bilinear": cv2.INTER_LINEAR,- "bicubic": cv2.INTER_CUBIC。

  • expand (bool，可选) - 是否要对旋转后的图片进行大小扩展，默认值：False。当参数值为 True 时，会对图像大小进行扩展，让其能够足以容纳整个旋转后的图像。当参数值为 False 时，会按照原图像大小保留旋转后的图像。这个扩展操作的前提是围绕中心旋转且没有平移。

  • center (tuple[int, int]，可选) - 旋转的中心点坐标，原点是图片左上角，默认值是图像的中心点。

  • fill (int，可选) - 对图像扩展时填充的值。默认值：0。

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

形状

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

  • output (PIL.Image|np.ndarray|Paddle.Tensor) - 返回随机旋转一定角度后的图像数据。

返回

计算 RandomRotation 的可调用对象。

代码示例

>>> import numpy as np
>>> from PIL import Image
>>> from paddle.vision.transforms import RandomRotation

>>> transform = RandomRotation(90)
>>> fake_img = Image.fromarray((np.random.rand(200, 150, 3) * 255.).astype(np.uint8))
>>> fake_img = transform(fake_img)
>>> print(fake_img.size)
(150, 200)