pytorch_debug
1、报错信息
1.1、出错位置
1 image = Image.open('./img.png')
2 # 图像预处理
3 transforms = transforms.Compose([transforms.Resize(256),
4 transforms.CenterCrop(224),
5 transforms.ToTensor(), # 转换为张量
6 # 归一化
7 transforms.Normalize([0.485, 0.456, 0.406],
8 [0.229, 0.224, 0.225])])
9 image = transforms(image)
1.2、最后一行报错
RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimension 0
1.3、解决方案
使用PIL.Image读取图像后，要用 .convert("RGB") 转为RGB三个通道。
image = Image.open('./img.png').convert("RGB")
