Pytorch中torchvision.transforms常用的数据变换

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/wolf2345/article/details/89012054

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图片数据,方便显示。

猜你喜欢

转载自blog.csdn.net/wolf2345/article/details/89012054
今日推荐