python - pytroch 数据加载器 RuntimeError: stack 期望每个张量大小相等，但在条目 0 处得到 [224, 224]，在条目 1 处得到 [224, 224, 3]

Question

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

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

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

并运行

for i in training_loader:
print(i)

我收到错误：

RuntimeError：堆栈期望每个张量大小相等，但在条目 0 处得到 [224, 224]，在条目 4 处得到 [224, 224, 3]

什么可能导致它以及如何解决它？先感谢您

Answer 1

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

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

有关更多详细信息，请参阅此答案。