pytorch入門學習:torchvision.transforms
阿新 • • 發佈:2019-02-10
torchvision.transforms是pytorch中的影象預處理包
一般用Compose把多個步驟整合到一起:
transforms.Compose([
transforms.CenterCrop(10),
transforms.ToTensor(),
])
這樣就把兩個步驟整合到一起
接下來介紹transforms中的函式:
-
Resize:把給定的圖片resize到given size
-
Normalize:Normalized an tensor image with mean and standard deviation
-
ToTensor:convert a PIL image to tensor (HW
-
ToPILImage: convert a tensor to PIL image
-
Scale:目前已經不用了,推薦用Resize
-
CenterCrop:在圖片的中間區域進行裁剪
-
RandomCrop:在一個隨機的位置進行裁剪
-
RandomHorizontalFlip:以0.5的概率水平翻轉給定的PIL影象
-
RandomVerticalFlip:以0.5的概率豎直翻轉給定的PIL影象
-
RandomResizedCrop:將PIL影象裁剪成任意大小和縱橫比
-
Grayscale:將影象轉換為灰度影象
-
RandomGrayscale:將影象以一定的概率轉換為灰度影象
-
FiceCrop:把影象裁剪為四個角和一箇中心
-
TenCrop
-
Pad:填充
-
ColorJitter:隨機改變影象的亮度對比度和飽和度