目录
一、Transforms是什么?
Transforms主要是对图片进行一些变换,一般用于数据预处理,数据在读取到pytorch之后通常都需要对数据进行预处理,包括尺寸缩放、转换张量、数据中心化或标准化等等,这些操作都是通过transforms进行的。
二、Transforms怎么用?
1.引入库
from torchvision import transforms
注:transforms.py就像是一个工具箱,里面有很多工具:
ToTensor()的使用:这个模块的主要作用主要就是将 PIL Image 或者numpy.ndarry 的图像类型转变成tensor的类型格式;例如将Image.open读出的数据就行转换,就需要用ToTensor()。
Resize()的使用:输入一张图片,将图片转变为指定的大小。如果只传入一个参数,就将最小边变成size,最大边等比例变成原来最小边的倍数。
2.构建自己的工具
工具箱里有很多工具,我们需要以某一个工具作为模具构建自己的工具,本质就是创建一个类的对象。例子如下:
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)