【pytorch】图像预处理torchvision.transforms

transforms进行图像预处理,Compose将多个步骤整合一起执行

transforms_=transforms.Compose([ 
            transforms.ToTensor(), # convert a PIL image to tensor (H*W*C) in range [0,255] to a torch.Tensor(C*H*W) in the range [0.0,1.0]
            normalize, # 用均值和标准差归一化张量图像
        ])

这两个基本是必须的

  1. ToTensor功能是将 PIL Image 类型 或者numpy.ndarray类型的图片对象转换为 tensor类型。
  2. normalize 则归一化张量

其他方法:

1.中心裁剪:

transforms.CenterCrop:中心裁剪
transforms.RandomCrop:随机裁剪
transforms.RandomResizedCrop:随机长宽比裁剪
transforms.FiveCrop:上下左右中心裁剪
transforms.TenCrop:上下左右中心裁剪后翻转

2.翻转和旋转

transforms.RandomHorizontalFlip(p=0.5):依概率p pp水平翻转
transforms.RandomVerticalFlip(p=0.5):依概率p垂直翻转
transforms.RandomRotation:随机旋转。

3.图像变换

transforms.Resize : 图像变换(resize)
transforms.Normalize:标准化
transforms.ToTensor: 转为tensor,并归一化为[0-1]
transforms.Pad :。填充
transforms.ColorJitter:修改亮度、对比度和饱和度
transforms.Grayscale:转灰度图
transforms.LinearTransformation():线性变换
transforms.RandomAffine:放射变换
transforms.ToPILImage:将数据转换为PILImage
transforms.Lambda:Apply a user-defined lambda as a transform
transforms.RandomGrayscale:依概率p pp转为灰度图

  • 5
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值