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")

更多推荐