pytorch学习笔记(四)——torchvision.transforms的使用

本文详细介绍了torchvision.transforms模块在图像预处理中的作用,包括ToTensor、Normalize、Resize和Compose等关键操作。通过实例展示了如何将图片转换为tensor,进行归一化处理,调整图片大小,并组合多个变换。此外,还演示了RandomCrop随机裁剪的使用方法,以及在tensorboard中展示图像变换效果的过程。

1. transforms作用

torchvision.transforms是pytorch中的图像预处理包,包含了很多种对图像数据进行变换的函数,例如裁剪翻转旋转等操作,这些都是在我们进行图像数据读入步骤中必不可少的。

2. transform各功能的用法

  1. ToTensor
    把图片数据转换成tensor数据类型
  2. Normalize
    归一化: Normalize创建实例时要输入均值和标准差,进行归一化操作输入需要是tensor类型的图片
  3. Resize
    缩放图片大小
  4. Compose
    一个transforms的系列变化,Compose中的参数是一个列表,列表元素要是transforms类型的,即Compose([transforms1,transforms2,…])
  5. RandomCop 随机裁剪
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

# python __call__函数的使用:直接用类对象(call的参数) 即可调用call函数

writer = SummaryWriter("logs")

img = Image.open("dataset/train/ants/5650366_e22b7e1065.jpg")

# 1. ToTensor 把图片数据转换成tensor数据类型
trans_totensor 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值