预处理之随机旋转

安全守望者 lv.1

发布时间：2023-02-12 01:49:55 380

$预处理之随机旋转_1024程序员节$

随机旋转：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()

预处理之随机旋转_旋转角度_02

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()

预处理之随机旋转_重置_03

预处理之随机旋转_旋转角度_04

配合使用，需要再重新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型变量
                                        ])

预处理之随机旋转_预处理_05

改变旋转中心

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()

预处理之随机旋转_旋转角度_06

文章来源： https://blog.51cto.com/u_15088375/5805517

特别声明：以上内容（图片及文字）均为互联网收集或者用户上传发布，本站仅提供信息存储服务！如有侵权或有涉及法律问题请联系我们。