預處理之随機旋轉

随機旋轉：transforms.RandomRotation

功能: 随機旋轉圖檔

主要參數說明：

1. degrees :旋轉角度

   當為a時，在(-a, a)之間選擇旋轉角度

   當為(a, b)時，在(a, b)之間選擇旋轉角度

2. resample :重采樣方法
3. expand :是否擴大圖檔，以保持原圖資訊
4. 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()