PIL Image转Pytorch Tensor
from torchvision import transforms
transform = transforms.Compose([
transforms.CenterCrop((512, 512)), # 只能对PIL图片进行裁剪
transforms.ToTensor(),
]
)
img= Image.open(img_path).convert('RGB')
im = transform (img)
im = im.unsqueeze_(dim=0) # C*H*W to B*C*H*W
img = Image.open(io.BytesIO(file_byte))