Pytorch中torchvision.transforms常用的數據變換

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

x^{^{normal}}=\frac{x-mean}{std}

也就是說(0,1)-0.5/0.5=(1-,1),即將數據歸一化爲[-1,1]的張量

(8)torchvison.transforms.ToPILImage:將Tensor變量數據轉換爲PIL圖片數據,方便顯示。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章