Pytorch问题---The size of tensor a (4) must match the size of tensor b (3) at non-singletonThe size of

报错:

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

解决:

img = Image.open(image_path) 改为
img = Image.open(image_path).convert('RGB')

完成~

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 游动-白 设计师:白松林 返回首页