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 相关文章推荐
在阿里云服务器上配置CentOS+Nginx+Python+Flask环境
Jun 18 Python
基于Python中求和函数sum的用法详解
Jun 28 Python
啥是佩奇?使用Python自动绘画小猪佩奇的代码实例
Feb 20 Python
如何使用Python进行OCR识别图片中的文字
Apr 01 Python
Python弹出输入框并获取输入值的实例
Jun 18 Python
Python整数对象实现原理详解
Jul 01 Python
Apache,wsgi,django 程序部署配置方法详解
Jul 01 Python
基于python的Paxos算法实现
Jul 03 Python
python中append实例用法总结
Jul 30 Python
Python闭包与装饰器原理及实例解析
Apr 30 Python
Python把图片转化为pdf代码实例
Jul 28 Python
Python  Asyncio模块实现的生产消费者模型的方法
Mar 01 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生成随机密码的三种方法小结
2010/09/04 PHP
php学习笔记之基础知识
2014/11/08 PHP
优化WordPress中文章与评论的时间显示
2016/01/12 PHP
非集成环境的php运行环境(Apache配置、Mysql)搭建安装图文教程
2016/04/12 PHP
php进行ip地址掩码运算处理的方法
2016/07/11 PHP
开发跨浏览器javascript常见注意事项
2009/01/01 Javascript
JS面向对象编程 for Cookie
2010/09/19 Javascript
精通Javascript系列之数据类型 字符串
2011/06/08 Javascript
JavaScript中的console.dir()函数介绍
2014/12/29 Javascript
JS实现的表格操作类详解(添加,删除,排序,上移,下移)
2015/12/22 Javascript
Bootstrap三种表单布局的使用方法
2016/06/21 Javascript
BootStrap实现带有增删改查功能的表格(DEMO详解)
2016/10/26 Javascript
解决微信内置浏览器返回上一页强制刷新问题方法
2017/02/05 Javascript
jquery ui 实现 tab标签功能示例【测试可用】
2019/07/25 jQuery
vue动态禁用控件绑定disable的例子
2019/10/28 Javascript
Vue.js页面中有多个input搜索框如何实现防抖操作
2019/11/04 Javascript
浅谈JSON5解决了JSON的两大痛点
2020/12/14 Javascript
[00:34]TI7不朽珍藏III——地穴编织者不朽展示
2017/07/15 DOTA
python spyder中读取txt为图片的方法
2018/04/27 Python
pytz格式化北京时间多出6分钟问题的解决方法
2019/06/21 Python
简单了解Django ContentType内置组件
2019/07/23 Python
Django 大文件下载实现过程解析
2019/08/01 Python
如何用OpenCV -python3实现视频物体追踪
2019/12/04 Python
基于Django实现日志记录报错信息
2019/12/17 Python
python打印异常信息的两种实现方式
2019/12/24 Python
使用TensorFlow对图像进行随机旋转的实现示例
2020/01/20 Python
selenium+python配置chrome浏览器的选项的实现
2020/03/18 Python
python ETL工具 pyetl
2020/06/07 Python
html5使用Canvas绘图的使用方法
2017/11/21 HTML / CSS
大学竞选班长演讲稿
2014/04/24 职场文书
感恩小明星事迹材料
2014/05/23 职场文书
大学生求职意向书
2015/05/11 职场文书
史上最牛辞职信
2015/05/13 职场文书
委托书范本格式
2019/04/18 职场文书
Go标准容器之Ring的使用说明
2021/05/05 Golang
Vue router配置与使用分析讲解
2022/12/24 Vue.js