Pytorch进行数据转换transforms

在构建dataset时,我们通常需要对数据进行转换,或者变换。

import torchvision.transforms as transforms
#compose相当于组合
my_transforms= transforms.Compose([
#转换为pillow图像
       transforms.ToPILImages(),
#随机剪裁
       transforms.RandomResizedCrop(224),
#水平翻转
       transforms.RandomHorizontalFlip(p=0.5),
#修改尺寸
       transforms.Resize(256),
#转换为tensor
       transforms.ToTensor(),
#标准化
       transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])

一般机器学习就用到这些,如果数据很少或者要求比较复杂可以参考官方文档。

你可能感兴趣的:(深度学习,python相关学习,图像处理,深度学习,人工智能,大数据)