torch.transforms

Posted 2021-03-02 lyl0618

在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图片数据，方便图片内容的显示