1. 程式人生 > >pytorch入門學習:torchvision.transforms

pytorch入門學習:torchvision.transforms

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

    C) in range [0,255] to a torch.Tensor(CHW) in the range [0.0,1.0]

  • 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:隨機改變影象的亮度對比度和飽和度