RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 ...

福芙芙_growing

已于 2022-09-01 10:01:51 修改

阅读量638

点赞数

CC 4.0 BY-SA版权

分类专栏：令人头大的环境问题文章标签： python

于 2022-04-07 21:00:38 首次发布

本文链接：https://blog.csdn.net/weixin_55768678/article/details/124026443

令人头大的环境问题专栏收录该内容

9 篇文章

订阅专栏

博客内容涉及在使用PyTorch训练模型时遇到的图像尺寸问题。问题一是transforms模块在调整图像尺寸时应使用正确的参数，例如对于Resnet模型，尺寸应为[224, 224]。问题二是数据集中图像通道不统一，存在RGB和灰度图像混杂的情况，解决方案是在加载图像时统一转换为RGB格式。通过修正这两个问题，可以避免RuntimeError并确保模型输入的正确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

两种可能：

一：transforms没有把数据集中的图片转换成想要的尺寸大小，如Resnet等网络输入需要224x224大小的图像，在Resize时，不能用 transforms.Resize(224)，而因该用transforms.Resize([224, 224])，官方文档有函数解释

二. 可能是因为自己的数据集中既有RGB图像也有灰度图像，通道不统一

参考https://discuss.pytorch.org/t/runtimeerror-invalid-argument-0-sizes-of-tensors-must-match-except-in-dimension-0-got-3-and-2-in-dimension-1/23890

可以通过以下方法解决：

在__getitem__() 方法中添加 image= Image.open(image).convert('RGB')

我这里是第一种问题，把size改成源码要求的大小就可以了

链接：https://www.jianshu.com/p/9e866d02ddbd