pytorch中torchvision的介绍

pytorch中torchvision.transforms.Compose()的介绍

1. torchvision.transforms

这就是一个对数据做预处理的工具包，可以对图像的数据做转换和裁剪大小等。

torchvision.transforms.Compose(arg):

其中arg是一个list类型，其中包含着所有你要对这个数据做的操作。

trans=[]

将数据转换成tensor类型
trans.append(transforms.ToTensor())

修剪图片大小
Resize（num） 这个通常在totensor之前，对PIL或NUmpy数据进行裁剪

转换图片格式到PIL
ToPIKImage

transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ，用均值（0.5, 0.5, 0.5）和标准差（0.5, 0.5, 0.5）对图像做归一化处理。


CentreCrop（）以输入图片的中心点做指定大小的裁剪
RandomCrop()选择一个随机点作为中心点进行裁剪

基本上你需要用什么方法就把它们组合成一个列表，然后构造Transformer：

transform = torchvision.transforms.Compose(trans)

其中trans是一个操作集合列表

例如利用pytorch导入手写数据集时，就可以加上你构造的transform，直接在读取的时候对数据进行转换：
mnist_train = torchvision.datasets.FashionMNIST(root=root, train=True, download=True, transform=transform)