pytorch 图像旋转

PyTorch是一个用于深度学习的开源机器学习库，它提供了许多用于图像处理的函数。要进行图像旋转，可以使用PyTorch中的torchvision.transforms模块中的函数来实现。

具体来说，要进行图像旋转，需要完成以下步骤：

1. 导入必要的库和模块：

   import torch
   from torchvision import transforms
   from PIL import Image

2. 创建一个transforms对象并使用其中的rotate函数来旋转图像。rotate函数可以接受一个旋转角度作为参数，例如：

   transform = transforms.Compose([
       transforms.ToTensor(),
       transforms.RandomRotation(degrees=30)
   ])

上面的代码将创建一个transforms对象，该对象将图像先转换为Tensor格式，然后随机旋转30度。

3. 加载要旋转的图像并应用transforms对象：

   img = Image.open('image.jpg')
   img_rotated = transform(img)

上面的代码将加载名为'image.jpg'的图像文件，并应用transforms对象来旋转该图像。

4. 将旋转后的图像转换回PIL格式并显示：

   img_rotated_pil = transforms.ToPILImage()(img_rotated)
   img_rotated_pil.show()

上面的代码将把旋转后的图像从Tensor格式转换回PIL格式，并显示出来。

希望这些信息对您有所帮助。