pytroch data loader RuntimeError: stack expects each tensor to be equal size, but got [224, 224] at entry 0 and [224, 224, 3] at entry 1

Question

我的问题是我在一个数据集中有两个张量 headers 图像和标签

当我执行简单循环时一切看起来都很好不幸的是，当我制作如下数据加载器时

training_loader = torch.utils.data.DataLoader(training_dataset, batch_size=100, shuffle=True)

和运行

for i in training_loader:
print(i)

我遇到错误：

RuntimeError: stack expects each tensor to be equal size, but got [224, 224] at entry 0 和 [224, 224, 3] 在入口 4

什么会导致它以及如何解决它？提前谢谢你

Answer 1

您的一张（或多张）图像似乎不是彩色图像，而是灰度图像。
修改您的加载代码以强制将所有图像视为彩色图像：

img = Image.load(filename).convert('RGB')

有关详细信息，请参阅。

python