process = transforms.Compose ([
#pytorch resize只支持PIL格式的图片,所以首先需要转成PIL
transforms.ToPILImage(),
#设置resize的图片尺寸
transforms.Resize(size=(224, 224)),
#将图片转为tensor
transforms.ToTensor(),
#归一化处理:[0-255] --> [0-1]
transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225]),
])
batch = process(crop)
return batch
详情见代码注释