pytorch_debug

1、报错信息

1.1、出错位置

image = Image.open('./img.png')
# 图像预处理
transforms = transforms.Compose([transforms.Resize(256),
                                                         transforms.CenterCrop(224),
                                                         transforms.ToTensor(),   # 转换为张量
                                                         # 归一化
                                                         transforms.Normalize([0.485, 0.456, 0.406],
                                                                                             [0.229, 0.224, 0.225])])
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")