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.NEAREST
和InterpolationMode.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
的示例