解决RuntimeError: stack expects each tensor to be equal size

于 2024-03-16 09:59:55 发布
阅读量925 收藏 11
点赞数 8
文章标签： python 目标检测 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/loinleeai/article/details/136756977
版权

问题报错：RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 256] at entry 0 and [1, 256, 256] at entry 9

这个问题是在我换数据集出现的

经过查看数据集，与原来用的数据集不同之处，我先把所有照片通道数尺寸都打印了

结果发现新的数据集ground truth是三通道，而原来是单个通道，把它们都转变成单通道就可以了。

此外，这个报错的根本原因就是每个batch要处理的数据张量尺寸是一样的才行，我的batch size大小为16，所以就是说第一个到第九个是 [3, 256, 256]，到了第十个成一个通道的，无法处理，导致报错

loinleeai
关注
  • 8
    点赞
  • 踩
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
RuntimeError: stack expects each tensor to be equal size, but got [3, ] at entry 0 and [1,]at entry1
yykkkybs的博客
05-26 1万+
RuntimeError: stack expects each tensor to be equal size, but got [3, 256, 256] at entry 0 and [1, 256, 256] at entry 1 该错误是我在用u-net（pytorch版）跑sar图像（灰度图）时爆出来的。 主要是因为设置的n_channel的值和读入图像img的实际通道数不符合，通过调试，我发现我的训练图像灰度图的通道数不全为1，里面还有3通道 于是就索性将dataset中读入图像的命令后
Python RuntimeError: thread.__init__() not called解决方法
09-22
主要介绍了Python RuntimeError: thread.__init__() not called解决方法,需要的朋友可以参考下
Torch数据加载维度不一致报错
weixin_43780880的博客
07-30 2193
@TOCTorch数据加载维度不一致报错 1、ERROR RuntimeError: stack expects each tensor to be equal size, but got [238, 128] at entry 0 and [55, 128] at entry 128。 2、解决方法 方案一： 数据预处理时，最好排除特征数据维度不一致的样本； 方案二： 如果实在无法排除，又在数据加载时报错，可以尝试： 1） 首先在get_item函数构建中判断数据维度，不符合目标维度的数据返回None类
解决pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\\python3.6\\lib\\site-packa的问题
01-21
标题用pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site-packages\numpy\init.py 1.file→settings→project interpreter→+（建议用pychon3.6版本，我之前用3.8版本安装不上numpy）， 2.搜索numpy,注意把下面对号点上 3.现在简单用numpy还是会报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码： with torch.no_grad(): # 停止自动反向计算梯度 参考： https://wisdomai.xyz/tool/pytorch/archives/2301 https://ptorch.com/news/160.html 作者：菜叶儿掉啦
在PyTorch学习过程中所遇问题
qq_45943487的博客
11-23 1497
PyTorch学习问题
深度学习中常见问题及知识点补充（持续更新中）
zhanlang619的博客
12-07 2537
深度学习中常见的代码运行问题，笔者在这里根据自己的学习情况进行了总结，持续更新中
Pytorch 中 Runtime Error & TypeError 之问题详解
゛flipped丶heart° 的博客
09-25 2309
当调用ImageFolder加载数据时，出现错误信息: TypeError: img should be PIL Image. 需要注意transform属性中Resize要在ToTensor前面！！！ 当调用ImageFolder加载数据时，出现错误信息: RuntimeError: stack expects each tensor to be equal size, but got []… 需要在transform中添加Resize（（224））属性，224不是绝对，具体看图片大小！！！ ..
调试分割网络模型代码错误记录
weixin_42487912的博客
08-25 2463
学习记录
RuntimeError: stack expects each tensor to be equal size
热门推荐
weixin_44394368的博客
10-22 2万+
在调试densenet进行分类任务的代码时，在图像预处理的过程中遇到下列错误： RuntimeError: stack expects each tensor to be equal size, but got [640, 640] at entry 0 and [560, 560] at entry 2 意思是我加载的张量大小不一致。 经过查找发现应该是我在加载图像时的预处理过程中出现了问题。 以下是训练数据预处理的实例化部分。 train_transform = Compose( [
RuntimeError: Cannot run the event loop while another loop is running(目前没有解决)
12-21
代码如下: import tornado.ioloop import tornado.web from tornado.httpclient import HTTPClient, AsyncHTTPClient class MainHandler(tornado.web.RequestHandler): # 同步 def get(self): h_c = HTTPClient() res = h_c.fetch("http://www.baidu.com") # print(res) # pass
RuntimeError: stack expects each tensor to be equal size, but got [1, 5] at entry 0 and [3, 5] at en
小帽哥的博客
07-15 8181
https://blog.csdn.net/yuxiang8546/article/details/118761999 在 dataset 类里 添加 def yolo_dataset_collate(batch): images = [] bboxes = [] for img, box in batch: images.append(img) bboxes.append(box) images = np.array(images)
RuntimeError: stack expects each tensor to be equal size, but got [3, 1053, 1872] at entry 0 and [3,
qq_45347185的博客
11-16 1万+
这个问题出现的原因是因为输入dataloader的图片大小并不一致。注意检查train, test, val数据集是否都进行了图片大小处理。如果没有的话，可以使用 test_transform = transforms.Compose([ transforms.Resize((256, 256)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.2
PyTorch代码错误笔记一
gengjuan520的博客
04-25 1572
PyTorch代码错误笔记 文章目录PyTorch代码错误笔记前言1、RuntimeError: stack expects each tensor to be equal size, but got [1, 224, 224] at entry 0 and [3, 224, 224] at entry 322、BrokenPipeError: [Errno 32] Broken pipe3、RuntimeError: CUDA out of memory. Tried to allocate 392.00
RuntimeError: stack expects each tensor to be equal size, but got at entry
Keeper的博客
04-02 4328
如果图像尺寸不固定一致的话，那batch_size设置为1，问题即可解决。
RuntimeError: stack expects each tensor to be equal size ？？
Lily的博客
03-29 2055
RuntimeError: stack expects each tensor to be equal size
深度学习中所遇问题
qq_45943487的博客
04-18 1463
原因：读取图片时类型错误，cv2.imread(path)返回numpy.ndarry类型，open(path,’rb’)返回_io.BufferedReader类型,使用PIL.Image.fromarray(cv2.cvColor(img. cv2.COLOR_BGR2RGB),同时又去安装了cv2。原因：花钱买教训，返回的batch中含None值，因为传入的label不能为None，将label值调为1后，无报错。原因：存在可能为None的image，返回batch数据会自动清理掉不合法的数据。
代码报错解决
最新发布
SCI211985的博客
10-19 411
超出了数据集的长度。遇到的错误，"stack expects each tensor to be equal size"，很可能是因为当批量大小超过数据集末尾时，会出现不同大小的张量。你的批量大小为12，如果在数据集末尾剩下的样本少于12个，你会得到形状为 [12, 6000] 和 [n, 6000] 的张量，其中 'n' 小于12。解决办法：在def __getitem__(self, index): 下添加以下代码。方法中添加的代码用于处理情况，其中。
RuntimeError: stack expects each tensor to be equal size, but got [0, 26] at entry 0 and [1, 26] at entry 1
07-11
这个错误通常是由于输入的张量形状不一致导致的。`torch.stack`函数要求输入的张量具有相同的形状，以便能够沿着指定的维度进行堆叠。 在您的情况下，错误信息显示了两个不同形状的张量：一个形状为 `[0, 26]` 的张量和一个形状为 `[1, 26]` 的张量。这意味着在堆叠这两个张量时，形状不匹配，因此引发了错误。 要解决这个问题，您需要确保在进行`torch.stack`操作之前，所有的输入张量具有相同的形状。您可以使用`torch.unsqueeze`函数在维度上扩展其中一个张量，使其与另一个张量的形状匹配，然后再进行堆叠。 以下是一个示例： ```python import torch # 假设有两个张量 tensor1 和 tensor2 tensor1 = torch.tensor([1, 2, 3]) tensor2 = torch.tensor([4, 5, 6]) # 在维度0上扩展 tensor1 的形状，使其与 tensor2 的形状匹配 tensor1 = torch.unsqueeze(tensor1, 0) # 进行堆叠操作 stacked_tensor = torch.stack([tensor1, tensor2], dim=0) ``` 在上述示例中，我们使用`torch.unsqueeze`函数在维度0上扩展了`tensor1`的形状，使其成为形状为 `[1, 3]` 的张量。然后，我们可以使用`torch.stack`将两个形状相同的张量堆叠在一起。 确保在使用`torch.stack`函数之前，所有输入的张量具有相同的形状，这样就可以避免出现错误。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值