1. transforms作用
torchvision.transforms是pytorch中的图像预处理包,包含了很多种对图像数据进行变换的函数,例如裁剪翻转旋转等操作,这些都是在我们进行图像数据读入步骤中必不可少的。
2. transform各功能的用法
- ToTensor
把图片数据转换成tensor数据类型 - Normalize
归一化: Normalize创建实例时要输入均值和标准差,进行归一化操作输入需要是tensor类型的图片 - Resize
缩放图片大小 - Compose
一个transforms的系列变化,Compose中的参数是一个列表,列表元素要是transforms类型的,即Compose([transforms1,transforms2,…]) - 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

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

被折叠的 条评论
为什么被折叠?



