def image_loader(image_name, max_sz=256):
""" forked from pytorch tutorials """
r_image = Image.open(image_name)
mindim = np.min((np.max(r_image.size[:2]), max_sz))
loader = transforms.Compose([transforms.CenterCrop(mindim),
transforms.ToTensor()])
image = Variable(loader(r_image))
return image.unsqueeze(0)
评论列表
文章目录