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


Posted in Python onFebruary 20, 2020

1、

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333

2、

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

bug1 修改方法是将图片resize 成一个尺度: 

解决1:

如果用transforms来resize ,关于尺寸的transform就是transforms.Resize(input_size, interpolation=3)了。

查看Resize的文档:

size (sequence or int) ? Desired output size. If size is a sequence like (h, w), output size will be matched to this.

If size is an int, smaller edge of the image will be matched to this number. i.e, if height > width, then image will be rescaled to (size * height / width, size)

size这个参数既可以是一个数字,又可以是一个tuple,如果是数字,图片处理后的输出尺寸每次都会被重新计算为(input_size*height/width, size),这样如果输入图片尺寸不一致,输出图片尺寸会不一致, 将导致bug1 问题

解决2:

也可以直接用opencv resize

以上这篇pytorch dataloader 取batch_size时候出现bug的解决方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
列举Python中吸引人的一些特性
Apr 09 Python
Python中关于使用模块的基础知识
May 24 Python
Python程序员面试题 你必须提前准备!(答案及解析)
Jan 23 Python
运用TensorFlow进行简单实现线性回归、梯度下降示例
Mar 05 Python
python将处理好的图像保存到指定目录下的方法
Jan 10 Python
python:按行读入,排序然后输出的方法
Jul 20 Python
Series和DataFrame使用简单入门
Nov 13 Python
python爬取王者荣耀全皮肤的简单实现代码
Jan 31 Python
Macbook安装Python最新版本、GUI开发环境、图像处理、视频处理环境详解
Feb 17 Python
Python中的Cookie模块如何使用
Jun 04 Python
用python-webdriver实现自动填表的示例代码
Jan 13 Python
python四个坐标点对图片区域最小外接矩形进行裁剪
Jun 04 Python
pytorch 使用加载训练好的模型做inference
Feb 20 #Python
pytorch中的inference使用实例
Feb 20 #Python
python encrypt 实现AES加密的实例详解
Feb 20 #Python
Python关于反射的实例代码分享
Feb 20 #Python
Python3监控疫情的完整代码
Feb 20 #Python
浅谈Pytorch torch.optim优化器个性化的使用
Feb 20 #Python
关于torch.optim的灵活使用详解(包括重写SGD,加上L1正则)
Feb 20 #Python
You might like
PHP类中Static方法效率测试代码
2010/10/17 PHP
PHP mysqli_free_result()与mysqli_fetch_array()函数详解
2016/09/21 PHP
PHP实现类似于C语言的文件读取及解析功能
2017/09/01 PHP
prototype Element学习笔记(Element篇三)
2008/10/26 Javascript
Prototype Class对象学习
2009/07/19 Javascript
firefox下对ajax的onreadystatechange的支持情况分析
2009/12/14 Javascript
jquery+php实现搜索框自动提示
2014/11/28 Javascript
jquery实现用户打分评分特效
2015/05/28 Javascript
JavaScript正则表达式之multiline属性的应用
2015/06/16 Javascript
bootstrap模态框消失问题的解决方法
2016/12/02 Javascript
Bootstrap模态框案例解析
2017/03/05 Javascript
vue如何从接口请求数据
2017/06/22 Javascript
react 实现页面代码分割、按需加载的方法
2018/04/03 Javascript
vue-cli3脚手架的配置及使用教程
2018/08/28 Javascript
浅析vue中的MVVM实现原理
2019/03/04 Javascript
vue 集成 vis-network 实现网络拓扑图的方法
2019/08/07 Javascript
基于Vue全局组件与局部组件的区别说明
2020/08/11 Javascript
python sys模块sys.path使用方法示例
2013/12/04 Python
Python实现的生成格雷码功能示例
2018/01/24 Python
python使用scrapy发送post请求的坑
2018/09/04 Python
python使用Plotly绘图工具绘制气泡图
2019/04/01 Python
Python generator生成器和yield表达式详解
2019/08/08 Python
Python模块future用法原理详解
2020/01/20 Python
selenium自动化测试入门实战
2020/12/21 Python
7款设计巧妙的css3飘带状3D立体效果的导航菜单和表单窗口
2013/02/04 HTML / CSS
HTML5拖拽文件上传的示例代码
2021/03/04 HTML / CSS
英国网上花店:Bunches
2016/11/29 全球购物
英智兴达软件测试笔试题
2016/10/12 面试题
北大自主招生自荐信
2013/10/19 职场文书
初中优秀班集体申报材料
2014/05/01 职场文书
邮政竞聘演讲稿
2014/09/03 职场文书
初级党校心得体会
2014/09/11 职场文书
2014入党积极分子批评与自我批评思想汇报
2014/09/20 职场文书
2014领导班子专题民主生活会对照检查材料思想汇报
2014/09/23 职场文书
2014学校领导四风对照检查材料思想汇报
2014/09/23 职场文书
工作表扬信
2015/01/17 职场文书