Pytorch 报错: The size of tensor a (4) must match the size of tensor b (3) at non-singleton

原因分析

1.读取的是.png 图片 是32位深度的,一般我们读取的是.jpg 24位深度的
2.转换为RGB就行了

test_image = Image.open(test_image_name).convert('RGB')

你可能感兴趣的:(报错信息,深度学习)