快捷方式

rotate

torchvision.transforms.functional.rotate(img: Tensor, angle: float, interpolation: InterpolationMode = InterpolationMode.NEAREST, expand: bool = False, center: Optional[List[int]] = None, fill: Optional[List[float]] = None) Tensor[source]

按角度旋转图像。如果图像是 torch Tensor,则其形状应为 […, H, W],其中 … 表示任意数量的前导维度。

参数:
  • img (PIL Image or Tensor) – 要旋转的图像。

  • angle (number) – 旋转角度值,以度为单位,逆时针方向。

  • interpolation (InterpolationMode) – 由 torchvision.transforms.InterpolationMode 定义的所需插值枚举。默认为 InterpolationMode.NEAREST。如果输入是 Tensor,仅支持 InterpolationMode.NEARESTInterpolationMode.BILINEAR。相应的 Pillow 整型常量,例如 PIL.Image.BILINEAR 也被接受。

  • expand (bool, optional) – 可选的扩展标志。如果为 true,则扩展输出图像使其足够大以容纳整个旋转后的图像。如果为 false 或省略,则输出图像与输入图像大小相同。请注意,expand 标志假定围绕中心旋转且没有平移。

  • center (sequence, optional) – 可选的旋转中心。原点是左上角。默认为图像中心。

  • fill (sequence or number, optional) –

    变换后图像区域外的像素填充值。如果给定一个数字,该值将用于所有波段。

    注意

    在 torchscript 模式下,不支持单个 int/float 值,请使用长度为 1 的序列:[value, ]

返回值:

旋转后的图像。

返回类型:

PIL Image 或 Tensor

使用 rotate 的示例

变换示例

变换示例

文档

获取 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源