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 Image 或 Tensor) – 要旋转的图像。
angle (数字) – 以度为单位的旋转角度值,逆时针方向。
interpolation (InterpolationMode) – 由
torchvision.transforms.InterpolationMode
定义的所需插值枚举。默认为InterpolationMode.NEAREST
。如果输入是 Tensor,则仅支持InterpolationMode.NEAREST
和InterpolationMode.BILINEAR
。也接受相应的 Pillow 整数常量,例如PIL.Image.BILINEAR
。expand (bool, 可选) – 可选的扩展标志。如果为 True,则扩展输出图像使其足够大以容纳整个旋转后的图像。如果为 False 或省略,则输出图像的大小与输入图像相同。请注意,扩展标志假定围绕中心旋转且没有平移。
center (序列, 可选) – 可选的旋转中心。原点是左上角。默认为图像中心。
fill (sequence 或 number, optional) –
变换后图像区域的像素填充值。如果提供数字,则该值将分别用于所有通道。
注意
在 torchscript 模式下,不支持单个 int/float 值,请使用长度为 1 的序列:
[value, ]
。
- 返回:
旋转后的图像。
- 返回类型:
PIL 图像或张量
使用
rotate
的示例