【pytorch】|torchvision transform

transforms类涵盖了大量对Tensor和对PIL Image的处理操作,是pytorch中的图像预处理包。

transforms.ToTensor()

将图像数据转换成了张量，并且归一化到了[0,1]。

transforms.normalize()

对张量进行归一化，它的形参包括mean、std等

对输入的tensor图像各个通道进行归一化

input[channel] = (input[channel] - mean[channel]) / std[channel]

transforms.Normalize( mean = (0.5,0.5,0.5), std = (0.5,0.5,0.5) )#将[0，1]的张量归一化到[-1, 1]上。

对于imagetNet ,平均值和方差是从imagenet训练集中抽样算出来的。
torchvision.transforms.Normalize(mean=[0.485,
0.456, 0.406], std=[0.229, 0.224, 0.225])

transforms.Compose

一般用Compose把多个步骤整合到一起：

transforms.Compose([
 
    transforms.CenterCrop(10),
 
    transforms.ToTensor(),
 
])