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