预处理之随机旋转

*Major*

已于 2022-10-24 13:49:53 修改

阅读量2.2k

点赞数 1

文章标签： 1024程序员节

于 2022-10-24 13:39:21 首次发布

本文链接：https://blog.csdn.net/qq_41375318/article/details/127490900

版权

该博客介绍了在图像预处理中如何使用`transforms.RandomRotation`进行随机旋转操作，包括旋转角度的选择、重采样方法、是否扩大图片以保持原图信息以及旋转中心的设置。通过实例展示了如何在Python中读取图像，应用预处理变换，并进行逆变换以展示预处理效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

$预处理之随机旋转$

随机旋转：transforms.RandomRotation

功能: 随机旋转图片

主要参数说明：

degrees :旋转角度
当为a时，在(-a, a)之间选择旋转角度
当为(a, b)时，在(a, b)之间选择旋转角度
resample :重采样方法
expand :是否扩大图片，以保持原图信息
center旋转点设置，默认中心旋转

from PIL import Image
from torchvision import transforms
from utils import transform_invert


if __name__ == '__main__':
    # 1.读取图像
    img = Image.open(r"./test.jpg").convert('RGB')
    # 2.确定预处理方式
    img_transform = transforms.Compose([transforms.Resize((300,300)),  # 重置大小为300*300
                                        transforms.RandomRotation(60),  # 在-60到60之间随机进行旋转
                                        transforms.ToTensor()  # 转Tensor型变量
                                        ])
    # 3.进行预处理
    img_tensor = img_transform(img)
    # 4.逆Transform变换
    img = transform_invert(img_tensor, img_transform)  # input: shape=[c h w]
    # 5.进行预处理效果展示
    img.show()

在这里插入图片描述

from PIL import Image
from torchvision import transforms
from utils import transform_invert


if __name__ == '__main__':
    # 1.读取图像
    img = Image.open(r"./test.jpg").convert('RGB')
    # 2.确定预处理方式
    img_transform = transforms.Compose([transforms.Resize((300,300)),  # 重置大小为300*300
                                        transforms.RandomRotation(90,expand=True),  # expand=True 会改变图像尺寸，来保持图像的完整性
                                        transforms.ToTensor()  # 转Tensor型变量
                                        ])
    # 3.进行预处理
    img_tensor = img_transform(img)
    # 4.逆Transform变换
    img = transform_invert(img_tensor, img_transform)  # input: shape=[c h w]
    # 5.进行预处理效果展示
    img.show()

在这里插入图片描述

配合使用，需要再重新Resize一下

    img_transform = transforms.Compose([transforms.Resize((300,300)),  # 重置大小为300*300
                                        transforms.RandomRotation(90,expand=True),  # 垂直翻转
                                        transforms.Resize((300, 300)),  # 重置大小为300*300
                                        transforms.ToTensor()  # 转Tensor型变量
                                        ])

在这里插入图片描述

改变旋转中心

from PIL import Image
from torchvision import transforms
from utils import transform_invert


if __name__ == '__main__':
    # 1.读取图像
    img = Image.open(r"./test.jpg").convert('RGB')
    # 2.确定预处理方式
    img_transform = transforms.Compose([transforms.Resize((300,300)),  # 重置大小为300*300
                                        transforms.RandomRotation(90,center=(0,0)),  # 改变旋转中心
                                        transforms.Resize((300, 300)),  # 重置大小为300*300
                                        transforms.ToTensor()  # 转Tensor型变量
                                        ])
    # 3.进行预处理
    img_tensor = img_transform(img)
    # 4.逆Transform变换
    img = transform_invert(img_tensor, img_transform)  # input: shape=[c h w]
    # 5.进行预处理效果展示
    img.show()