什么是Transforms
在PyTorch中,transforms是用于对数据进行预处理、增强和变换的操作集合。transforms通常用于数据载入和训练过程中,可以包括数据的归一化、裁剪、翻转、旋转、缩放等操作,以及将数据转换成PyTorch可以处理的Tensor格式。
Transforms的使用
首先导入包
from torchvision import transforms
实际上是导入了一个文件,文件名为transforms,我们目前主要查看一下其中的ToTensor类
我们可以查看一下其中的源码

我们可以看到这个类的主要作用是 将PIL Image或者是numpy.ndarray类型转化为tensor类型
其中的__call__方法,类似于c++重载()运算符

使用PIL image格式
PyTorchTransforms:数据转换与ToTensor操作指南,

最低0.47元/天 解锁文章
1662

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



