【深度学习】图片张量错误 The size of tensor a (4) must match the size of tensor b (3) at non-singleton
Pytorch 写 Dataset 的时候遇到的这个问题The size of tensor a (4) must match the size of tensor b (3) at non-singleton解决方法Image.open(image)改成Image.open(image).convert("RGB")
·
Pytorch 写 Dataset 的时候遇到的这个问题
The size of tensor a (4) must match the size of tensor b (3) at non-singleton
解决方法
Image.open(image)
改成
Image.open(image).convert("RGB")
更多推荐
所有评论(0)