pytroch 数据加载器运行时错误：堆栈期望每个张量的大小相等，但在条目 0 处得到 [224, 224]，在条目 1 处得到 [224, 224, 3]

托马斯·卡茨马尔斯基

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

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

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

并运行

for i in training_loader:
print(i)

我收到错误：

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

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

沙伊

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

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

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

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。

编辑于 2021-09-7

0 条评论

登录后参与评论

TOP 榜单