快捷方式

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[源代码]

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

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

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

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

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

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

  • fill (sequencenumber, optional) –

    变换后图像区域的像素填充值。如果提供数字,则该值将分别用于所有通道。

    注意

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

返回:

旋转后的图像。

返回类型:

PIL 图像或张量

使用 rotate 的示例

转换图示

转换图示

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源