torchvision.transforms中常用的數據變換
(1)torchvision.transforms.Resize():用於對載入的圖片數據按我們需求的大小進行縮放。傳遞給這個類的參數可以使一個整形數據,也可以是一個類似於(h,w)的序列,其中,h代表高度,w代表寬度,但是如果使用的是一個整形數據,那麼代表縮放的寬度和高度都是這個整形數據的值。
(2)torchvision.transforms.Scale:用於對載入的圖片數據按我們需求進行縮放,和torchvision.transforms.Resize()用法類似。
(3)torchvision.transforms.CenterCrop:用於對載入圖片以圖片中心爲參考點,按照需求進行裁剪。
(4)torchvision.transforms.RamdomCrop:用於對載入的圖片按需求進行隨機裁剪。
(5)torchvision.transforms.RandomHorizontalFlip:對載入圖片按隨機概率進行水平翻轉,默認概率值爲0.5.。
(6)torchvision.teansforms.RandomVerticalFlip:對載入圖片按隨機概率進行垂直飯莊,默認概率值爲0.5。
(7)torchvision.transoforms.ToTensor:對載入圖片進行類型轉換,將PIL圖片數據轉換爲Tensor數據類型數據。
ex:transform=transforms.Compose([transform.ToTensor(),transforms.Normalize(mean=[0.5,0.5,0.5],std=[0.5,0.5,0.5])])
也就是說(0,1)-0.5/0.5=(1-,1),即將數據歸一化爲[-1,1]的張量
(8)torchvison.transforms.ToPILImage:將Tensor變量數據轉換爲PIL圖片數據,方便顯示。