pytroch数据加载器RuntimeError:堆栈期望每个张量大小相等，但条目0处为[224,224]，条目1为[224,224，3]。
pytroch数据加载器RuntimeError:堆栈期望每个张量大小相等，但条目0处为[224,224]，条目1为[224,224，3]。
提问于 2021-08-05 09:41:57
我的问题是，在一个数据集中有两个张量，头、图像和标签。
当我执行简单的循环时，当我让dataloader如下所示时，一切看起来都很糟糕。
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
是什么导致了它，以及如何解决它？先谢谢你
回答 1
Stack Overflow用户
回答已采纳
发布于 2021-08-05 10:01:59
看起来你的一个(或多个)图像不是彩色图像，而是灰度图像。
修改加载代码，强制将所有图像视为彩色图像：
img = Image.load(filename).convert('RGB')有关更多详细信息，请参阅this answer。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/68664217
复制相关文章
相似问题
腾讯云开发者
