RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 ...
两种可能:
一: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
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 ...相关推荐
- 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.
报错信息: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 186 a ...
- 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 ...
- 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:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
数据加载错误导致 可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一样的问题,但是这里没问题 我又查看了我的trans ...
- pytorch报错: invalid argument 0: Sizes of tensors must match except in dimension 0.
这个问题是在dataloader遇到的,网上一般是数据的维度不同,但我这里其实是最后一份数据和之前的数据比数量不足,所以凑不够一个batch,所以我的做法是加入drop_last=True意思是如果数 ...
- RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.
最近在修改模型不支持的算子后,由于对模型的效果产生了影响,所以需要进行微调训练,但是训练的时候报错了,如下: File "/home/xxxxxxx/EdgeNets/nn_layers/e ...
- 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 ...
- 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, ...
最新文章
- html5页面输出语句,使用html5输出文件
- 使用Freestreamer 拖动滑动条音频重新播放或者没有声音seekToPosition无效
- jq的form data中保留加号_使用Fourier+EMD水管中传播的单频音波进行滤波(Python)...
- centos7配置 console口_玩转KVM-一招打开vm的console口
- NFS 服务器配置说明
- 【Boost】boost库中智能指针——scoped_ptr
- mysql 创建账号权限_mysql创建用户,并指定用户的权限(grant命令)
- IOS之代理文字点击变大变小
- easyui 排序实现
- 你想象中的实习是什么样的?
- python中字符编码使用_python中字符编码是什么?如何转换字符?
- 千万不要这样写代码!9种常见的OOM场景演示
- Python让繁琐工作自动化——chapter12 处理Excel电子表格
- 织梦mysql安装教程视频教程_dedecms织梦模板安装教程视频/图文步骤(模板秀出品)...
- Software System [vaynexiao]
- LINUX-AWK-删除首行、删除尾行、删除首尾两行
- 墨卡托坐标转GPS坐标
- 微众银行马智涛:解放数据生产力,以金融科技助力数字经济发展|数字思考者50人...
- GRAIL Efficient Time Series Representation Learning论文阅读笔记(三)
- 风暴孵化:手游代理加盟水很深,如何避免入坑?
热门文章
- 昆仑通态如何连接sqlserver数据库_mcgs通用和嵌入的区别
- 【Web前端】HTMLCSS一文详解
- 应用之星:制作H5婚礼邀请函,让你的婚礼与众不同!
- 搜狗公交:出行在左,谋略在右
- 京东视觉算法部招聘CV实习生
- 7744问题的两种解法
- MQ-5 气体传感器对接巴法云发送微信消息
- 智慧树python数据分析与数据可视化章节答案_知到智慧树_Python数据分析与数据可视化_章节测验答案...
- python实战故障诊断之CWRU数据集(三):信号预白化处理-倒谱预白化(CEP pre-whitening)
- java头像_用java实现给你的头像) +n