PyTorch中的图像旋转：torchvision.transforms.functional.rotate详解

PyTorch中的图像旋转：torchvision.transforms.functional.rotate详解

在深度学习和计算机视觉任务中，图像预处理是一个非常重要的步骤。其中，图像旋转是一种常见的预处理操作，可以用于数据增强、对齐图像等。PyTorch的torchvision库提供了许多用于图像预处理的工具，其中transforms.functional.rotate函数可以用于对图像进行旋转。

函数基本用法

torchvision.transforms.functional.rotate函数的基本用法如下：

torchvision.transforms.functional.rotate(img, angle, interpolation=2, expand=False, center=None, resample=False, fillcolor=0)

参数解释：

• img：待旋转的图像，通常是一个PIL Image对象或NumPy ndarray。
• angle：旋转角度，以度为单位，逆时针为正方向。
• interpolation：插值方法，用于旋转后的像素插值。默认为2，表示双线性插值（bilinear）。
• expand：是否扩展输出图像以包含旋转后的完整图像。默认为False，即旋转后的图像会被裁剪以适应原始图像大小。
• center：旋转中心，默认为图像中心。可以是一个二元组(x, y)，表示旋转中心的坐标。
• resample：与interpolation参数类似，用于指定插值方法。该参数已被弃用，建议使用interpolation参数。
• fillcolor：当expand为True且旋转后的图像超出原始图像大小时，用于填充新区域的颜色值。默认为0，表示黑色。

实际应用示例

下面是一个使用torchvision.transforms.functional.rotate函数对图像进行旋转的示例：

import torch
from PIL import Image
from torchvision import transforms
# 加载图像
img = Image.open('example.jpg')
# 定义旋转角度
angle = 45
# 创建旋转变换
rotate_transform = transforms.functional.rotate
# 对图像进行旋转
rotated_img = rotate_transform(img, angle)
# 显示旋转后的图像
rotated_img.show()

在这个示例中，我们首先使用PIL库加载了一张图像。然后，我们定义了旋转角度为45度，并创建了一个旋转变换对象rotate_transform。最后，我们使用rotate_transform函数对图像进行旋转，并将旋转后的图像显示出来。

注意事项

• 当expand参数为False时，旋转后的图像会被裁剪以适应原始图像大小。如果希望保留旋转后的完整图像，需要将expand设置为True。
• fillcolor参数仅在expand为True时有效。当expand为True且旋转后的图像超出原始图像大小时，新区域将使用fillcolor指定的颜色进行填充。
• resample参数已被弃用，建议使用interpolation参数来指定插值方法。

通过以上介绍，相信你已经对torchvision.transforms.functional.rotate函数有了更深入的了解。在实际应用中，你可以根据需要对图像进行旋转操作，以实现数据增强、图像对齐等目的。希望本文能对你有所帮助！