快捷方式

affine

torchvision.transforms.functional.affine(img: Tensor, angle: float, translate: List[int], scale: float, shear: List[float], interpolation: InterpolationMode = InterpolationMode.NEAREST, fill: Optional[List[float]] = None, center: Optional[List[int]] = None) Tensor[source]

对图像应用仿射变换,保持图像中心不变。如果图像为 Torch 张量,则预期其形状为 […, H, W],其中 … 表示任意数量的前导维度。

参数:
  • img (PIL ImageTensor) – 要变换的图像。

  • angle (数字) – 旋转角度,以度为单位,介于 -180 到 180 之间,顺时针方向。

  • translate (Python 整数序列) – 水平和垂直平移(旋转后平移)。

  • scale (浮点数) – 总体缩放比例。

  • shear (浮点数序列) – 剪切角值,以度为单位,介于 -180 到 180 之间,顺时针方向。如果指定序列,则第一个值对应于平行于 x 轴的剪切,而第二个值对应于平行于 y 轴的剪切。

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

  • fill (序列数字, 可选) –

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

    注意

    在 TorchScript 模式下,不支持单个整数/浮点值,请使用长度为 1 的序列:[value, ]

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

返回值:

变换后的图像。

返回类型:

PIL Image 或 Tensor

使用 affine 的示例

变换的说明

变换的说明

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源