pytorch 旋转

在 PyTorch 中，你可以使用 torchvision.transforms 模块中的 transforms 来实现图像的旋转。transforms 提供了一系列的图像变换函数，包括旋转、缩放、裁剪等操作。

要对图像进行旋转，可以使用 transforms.RandomRotation 函数。它会随机地对图像进行旋转，可以指定旋转角度的范围。下面是一个示例代码：

import torchvision.transforms as transforms
from PIL import Image

# 加载图像
image = Image.open('image.jpg')

# 定义旋转角度范围
angle = 45

# 定义旋转变换
transform = transforms.RandomRotation(angle)

# 对图像进行旋转
rotated_image = transform(image)

# 显示旋转后的图像
rotated_image.show()

上述代码中，我们首先加载了图像，然后定义了旋转角度范围为 -45 到 45 度。接下来，我们创建了一个 RandomRotation 变换对象，并将其应用于图像上。最后，我们显示了旋转后的图像。

希望这个示例对你有帮助！如有任何问题，请随时提问。