RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1and 3 解决方法

最新推荐文章于 2025-05-21 11:59:08 发布

转载最新推荐文章于 2025-05-21 11:59:08 发布 · 4.5k 阅读

4 ·

CC 4.0 BY-SA版权

原文链接：https://blog.csdn.net/weixin_41278720/article/details/84586734

文章标签：

#convert #pytorch #image

Python Debug 同时被 2 个专栏收录

9 篇文章

订阅专栏

Pytorch

4 篇文章

订阅专栏

本文探讨了在使用PyTorch的DataLoader加载图像数据时遇到的RuntimeError：invalid argument错误，该错误源于图像通道数不一致导致批次构建失败。文章提供了解决方案，即在读取图像时统一转换为RGB格式，确保所有图像具有相同的通道数。

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

pytorch报错

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 and 3 in dimension 1 at /pytorch/aten/src/TH/generic/THTensorMath.cpp:3616

原因分析

使用DataLoader加载图像，这些图像中的一些具有3个通道（彩色图像），而其他图像可能具有单个通道（BW图像），由于dim1的尺寸不同，因此无法将它们连接成批次。

解决方法

将img = img.convert（‘RGB’）添加到数据集图像读取中

img = Image.open(img_path)
img = img.convert('RGB')

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

洞幺01

关注关注

0
点赞
踩
4

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

专栏目录

invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1

jacke121的专栏

01-28

1万+

invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 原因：dataloader 需要的数据维度是一样的，有的有两个目标，有的有一个目标，就会出现异常：解决方法：凑成一样的。 Traceback (most recent call last): File "G:/project/f...

解决问题使用invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1

牛肉胡辣汤

11-28

728

在这个示例中，我们首先检查了特征张量和标签张量的尺寸，确保它们匹配。在神经网络中，各个层之间的输入和输出张量的尺寸必须匹配，以确保各层之间的连接正确。在使用张量进行计算的过程中，我们需要经常检查和调整张量的尺寸，以确保它们与其他张量的尺寸匹配。如果我们确定张量的尺寸是正确的，并且我们希望进行不同尺寸的张量操作，那么我们可以使用广播机制来解决这个问题。张量的尺寸可以用来描述张量在每个维度上的大小以及它们之间的关系。在这个例子中，由于广播机制的作用，我们可以成功地对这两个不同尺寸的张量进行相乘操作。

参与评论您还未登录，请先登录后发表或查看评论

Pytorch 多卡训练报错 RuntimeError: one of the variables needed for gradient computation has been modified

最新发布

既脱春火的博客

05-21

289

查了很多资料，都说是in_place的问题，主要是在relu以及 +=、*=的问题上，再有就是pytorch版本（pytorch内置的模型中会出现in_place问题）；relu的问题还能通过下面这个函数替换参数解决，+=、*=这种问题好像只能通过修改代码实现。主要是添加broadcast_buffers=False这个参数，问题解决了。

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

sunflower_sara的机器学习园地

08-23

1万+

报错如下： RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 4 in dimension 1 at /pytorch/aten/src/TH/generic/THTensorMath.cpp:3616 因为读取图像的时候三通道图像中混入了灰度或...

pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 a

公众号/知乎: AI算法札记，欢迎关注交流

11-28

3万+

1.pytorch报错 Namespace(batchSize=16, gpu='0,1', lr=0.001, nepoch=210, num_workers=8) Epoch: 0 Epoch:0 [0|173089] loss:3.747539 Epoch:0 [1|173089] loss:3.629151 Epoch:0 [2|173089] loss:3.731266 Traceba...

pytorch dataload 出错：: invalid argument 0: Sizes of tensors must match except in dimension 0

02-28

836

错误信息：”RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 299 and 252 in dimension 3 at /pytorch/aten/src/TH/generic/THTensor.cpp:714 错误地方原代码： transforms.Resize(...

RuntimeError: Sizes of tensors must match except in dimension 1. Got 16 and 32 in dimension 2报错解决

m0_57814193的博客

05-11

2131

比如，如果你有两个特征图，一个是 16×16，另一个是 32×32，那么你可以将 16×16 的特征图上采样到 32×32。在大多数情况下，当使用 YOLO 或类似的体系结构时，这种错误通常发生在特征图融合阶段，即当尝试将来自不同层级的特征图进行拼接时。通常，特征图拼接是沿着特征通道维度进行的（即维度1，对应PyTorch中的通道），但你需要确保所有操作前后都保持这一约定。在你的网络中，如果你在拼接不同的特征层（可能是不同的卷积层输出），你需要确保这些特征图在拼接前在高度和宽度上是匹配的。

Sizes of tensors must match except in dimension 0. Got 339 and 387 问题解决

qq_38376205的博客

12-02

2722

Training is started. Traceback (most recent call last):

pytorch dataloader 取batch_size时候出现bug的解决方式

12-20

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

深度学习Pytorch-常见问题(二)

yanzhiwen2的博客

04-14

2673

深度学习Pytorch-常见问题(二)

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

agonysome的博客

10-28

1043

查到了两个解决方案，其一如下，经过查验不是这个原因造成了我的错误：原链接：https://www.cnblogs.com/zxj9487/p/11531888.html 解决：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 1935 in dimension 2 at ../aten/src/TH/generic/THTensor.cpp:711 这种错误

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.

Lily's Blog

05-25

1万+

pytorch训练网络模型时报错： RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.Got 160 and 156 in dimension 2 at xxxxx. 上网搜索了一下，参考https://www.cnblogs.com/zxj9487/p/11531888.html文章所述： “1. 你输入的图像数据的维度不完全是一样的；2.batchsize的整除问题（总数据...

RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 58 but got size 57 fo

热门推荐

梦在黎明破晓时的博客

11-13

3万+

在使用yolo v5n进行网络训练时使用的，在第12层Concat处连接第11层和第6层的Feature Map时出现维度不同的问题。

RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2

Guo_Python的博客

01-15

2万+

RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2

RuntimeError: torch.cat(): Sizes of tensors must match except in dimension 1. Got 48 and 16 in dimen

weixin_44466434的博客

05-17

2324

具体来说，就是将原来一个低分辨的像素划分为rr个更小的格子，利用rr个特征图对应位置的值按照一定的规则来填充这些小格子。C=torch.cat((A,B),0)就表示按维数0（行）拼接A和B，也就是竖着拼接，A上B下。C=torch.cat((A,B),1)就表示按维数1（列）拼接A和B，也就是横着拼接，A左B右。.cat是将两个张量（tensor）拼接在一起，cat是concatnate的意思，即拼接，联系在一起。是其属性，而size()是其继承的方法，两者均可以获得tensor的维度。

记录解决RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 27 but got size

h363924219的博客

11-09

1万+

在做目标检测服务过程中，将yolov7模型通过flask打包成预测服务API，此次训练的图像输入大小是1280，输入预测图片是如果图像大于1280则预测成功，小于1280则报RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 27 but got size。由于只有小图片预测报错，猜测是图像处理过程中resize问题，提示下面代码行错误。

解决报错：RuntimeError: Sizes of tensors must match except in dimension 1.

weixin_44415029的博客

11-12

3304

仔细检查后发现，Controlnet的这个pipe对输入要求很严格。其次，图像的格式方面也有要求，必须是RGB格式。但是，我统一尺寸之后（如下），这个报错仍然存在。这里会出问题是我没想到的。

DataLoader加载图像时Sizes of tensors must match except in dimension 0. Got 1and 3 in dimension 1的问题解决方法

dcxhun3的博客

10-21

1461

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 and 3 in dimension 1 at /opt/conda/conda-bld/pytorch_1579040055865/work/aten/src/TH/generic/THTensor.cpp:612 通常在pytorch环境中，DataLoader加载图像时，用的是 Image库读取图像，代码如下： fro.

RuntimeError: Sizes of tensors must match except in dimension 1. Got 3 and 1 (The offending index is 0)

05-10

这个错误通常出现在两个张量的形状（shape）不匹配时进行操作时。在维度1以外，这两个张量的形状必须完全相同。但是在维度1处，它们的长度可以不同。根据错误信息，你正在进行的操作涉及到两个张量，其中一个张量在维度1处的长度为3，而另一个张量在维度1处的长度为1。这是不允许的，两个张量在维度1处的长度必须相同。你需要检查你的代码，找出哪个张量在维度1处的长度与另一个张量不同，并相应地修改它们的形状，使它们能够进行操作。你可以使用 PyTorch 中的 `view()` 或 `reshape()` 函数来改变张量的形状。