pytorch:data读取出错:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension
在使用Dataloader读取数据的时候,使用batch_size=1不会出现这个问题。当batch_size>1时, 默认将会使用torch.stack()为你生成一个[batch,x, x, x] 的tensor数据,在使用该函数时需要输入的两个tensor维度一样。
1、注意自己图像的大小是否resize到相同尺寸。
2、图像的通道数是否相同,全彩色或全灰度,使用相同的类型。
3、如果是CNN的多标签分类,标签长度不同也可以用下面的办法。
3、在进行目标检测的时候,我们返回的不仅仅是图像数据,还有它的gt_box以及gt_label。但是每个图像的标签个数不是相同的,所以在使用Dataloader原始的参数时也会报错。此时可以自己写一个 collate_fn函数,因为我的返回为图像、目标框以及类别标签所以修改如下
def data_collate(batch):gt_box = []gt_label = []imgs = []for info in batch:imgs.append(info[0])gt_box.append(info[1])gt_label.append(info[2])return torch.stack(imgs, 0), gt_box,gt_label
train里面调用:DataLoader(data_mine, batch_size=2,collate_fn=data_collate, shuffle=True, num_workers=2)
根据自己重写的Dataset类的返回值修改上面,得到自己想要的数据。
pytorch:data读取出错:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension相关推荐
- pytorch dataload 出错:: invalid argument 0: Sizes of tensors must match except in dimension 0
错误信息:"RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got ...
- Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
报错信息: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 186 a ...
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.
最近在修改模型不支持的算子后,由于对模型的效果产生了影响,所以需要进行微调训练,但是训练的时候报错了,如下: File "/home/xxxxxxx/EdgeNets/nn_layers/e ...
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0—问题分析
声明: 这里仅仅提供一个思路,暂时没有绝对正确通用的解决方案 作者:{ 墨理 } 感谢通过私信.评论交流,协力探讨解决问题的同学 UP .Penser 文章目录 探讨的 pytorch 报错信息如下 ...
- pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
数据加载错误导致 可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一样的问题,但是这里没问题 我又查看了我的trans ...
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 ...
两种可能: 一:transforms没有把数据集中的图片转换成想要的尺寸大小,如Resnet等网络输入需要224x224大小的图像,在Resize时,不能用 transforms.Resize(224 ...
- invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1
invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 原因:dataloader 需要的数据维度是一 ...
- pytorch报错: invalid argument 0: Sizes of tensors must match except in dimension 0.
这个问题是在dataloader遇到的,网上一般是数据的维度不同,但我这里其实是最后一份数据和之前的数据比数量不足,所以凑不够一个batch,所以我的做法是加入drop_last=True意思是如果数 ...
- Pytorch_Geometric(PyG)使用DataLoader报错RuntimeError: Sizes of tensors must match except in dimension 0.
使用Pytorch_Geometric(PyG)时构建DataLoader,从DataLoader获取样本Batch时报错:RuntimeError: Sizes of tensors must ma ...
最新文章
- 计算机丢失cv210.dll,cv210.dll文件免费版
- Android - Glide4.4.0使用
- github初学者指南_GitHub初学者指南
- php生成随机不重复的数字_PHP生成不重复随机数的几种方法
- 消控中心人员配置_电气火灾监控系统在石药集团新药制剂配套特色原料药FDA生产中心项目...
- docker~aspnetcore2.0镜像缺少libgdiplus问题
- Java知多少(9) import及Java类的搜索路径
- 笔记本电脑风扇控制软件
- 样本量估算:随机对照试验(两组均数)比较的样本量计算方法
- PyQt5 基本语法(一):基类控件
- 【Unity3D游戏开发实战】Unity3D实现休闲类游戏《2048》——算法、源代码
- 逻辑运算符 异或是什么意思 异或运算
- Arduino使用u8g2库函数驱动4线/6线OLED屏幕(I2C/SPI通讯)附带库函数详解
- 收藏!5款超级好用的小工具推荐,各个都是同类软件中的翘楚
- html五子棋人机对战,JS+canvas实现五子棋人机大战
- 第七届省赛-海岛争霸(最短路Floyd)
- PowerSync 包尔星克 (群加) 对拷线/共享线连不上
- 腾讯大逆转阿里,夺回港股王,马化腾“拍了拍”马云
- mysql 获取倒数第二_MySQL查询倒数第二条记录实现方法
- 华硕系统安装程序正在启动服务器,华硕电脑用u盘如何装系统win7