Fork me on GitHub 0

torch.transforms

在torchvision.transforms中常用的数据变换操作:

torchvision.transforms.Resize:对图片数据按需求大小进行缩放,传递的参数为整型,(h,w)h表示高度,w表示高度

torchvision.transforms.Scale:对图片按需求大小进行缩放

torchvision.transforms.CenterCrop:对图片以图片中心为参考点,按需求大小进行裁剪

torchvision.transforms.RandomCrop:对图片按需求大小进行随机裁剪

torchvision.transforms.RandomHorizontalFlip:对图片按随机概率进行水平翻转,概率默认值为0.5

torchvision.transforms.RandomVerticalFlip:对图片按随机概率进行垂直翻转,概率默认值为0.5

torchvision.transforms.ToTensor:对图片进行类型转换,之前构成PIL图片的数据转换成Tensor数据类型的变量,让Pytorch能够进行计算和处理

torchvision.transforms.ToPILImage:将Tensor变量的数据转换为PIL图片数据,方便图片内容的显示

 

posted @ 2020-04-07 18:45  amazingcode  阅读(1284)  评论(0编辑  收藏  举报