transforms提供了对PIL Image对象和Tensor对象的常用操作
ToTensor
将一个取值范围在[0,255]的PIL.Image转换为Tensor。形状为(H,W,C)的Numpy.ndarry转换为形状[C,H,W],取值范围是[0,1.0]的torch.FloatTensor。
Normalize
即进行标准化操作。
image = (image - mean)/ std
mean : 均值
std :标准差
其中mean与std均人为设置
实例
trans = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize([0.5], [0.5])])