torchvison.transforms.functional.rotate
时间: 2024-09-29 22:05:10 浏览: 39
`torchvision.transforms.functional.rotate` 是 PyTorch 中 torchvision 模块下的一个功能函数,它主要用于图像数据增强。这个函数用于旋转输入的 PIL Image 对象,使得图像围绕其中心点按指定的角度顺时针或逆时针旋转。
函数原型通常是这样的:
```python
rotate(image: PIL.Image, angle: int, resample: int = PIL.Image.BILINEAR, expand: bool = False, fill: tuple = (0, 0), center: Optional[Tuple[int, int]] = None) -> PIL.Image
```
参数说明:
1. `image`: 需要旋转的 PIL Image 对象。
2. `angle`: 要旋转的角度,单位是度。
3. `resample`: 可选插值模式,默认为 PIL.Image.BILINEAR,表示双线性插值,其他选项如 NEAREST、BICUBIC 等也支持。
4. `expand`: 是否保持原始尺寸并填充背景颜色,默认为 False。
5. `fill`: 当 `expand` 为 True 时,新区域使用的填充颜色,默认为黑色(0, 0)。
6. `center`: 可选旋转中心点,如果未提供,则默认为图像的中心。
使用该函数时,你需要先安装 torchvision 库,并通过 PIL 库加载图片,然后调用 `rotate` 函数对图片进行旋转操作。这是基本的一个例子:
```python
from PIL import Image
import torchvision.transforms as transforms
# 加载图片
img = Image.open('your_image.jpg')
rotated_img = transforms.functional.rotate(img, 90)
```
阅读全文