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在命令行下使用google翻译(带语音)
Jan 16 Python
python读取csv文件示例(python操作csv)
Mar 11 Python
安装dbus-python的简要教程
May 05 Python
Eclipse和PyDev搭建完美Python开发环境教程(Windows篇)
Nov 16 Python
Python最火、R极具潜力 2017机器学习调查报告
Dec 11 Python
python 定义给定初值或长度的list方法
Jun 23 Python
Python计算库numpy进行方差/标准方差/样本标准方差/协方差的计算
Dec 28 Python
python使用beautifulsoup4爬取酷狗音乐代码实例
Dec 04 Python
python实现简单俄罗斯方块
Mar 13 Python
django中url映射规则和服务端响应顺序的实现
Apr 02 Python
Django配置跨域并开发测试接口
Nov 04 Python
Python可变集合和不可变集合的构造方法大全
Dec 06 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
星际争霸任务指南——虫族
2020/03/04 星际争霸
php中通过Ajax如何实现异步文件上传的代码实例
2011/05/07 PHP
PHP图片处理之使用imagecopyresampled函数实现图片缩放例子
2014/11/19 PHP
PHP中使用Imagick操作PSD文件实例
2015/01/26 PHP
PHP实现QQ登录实例代码
2016/01/14 PHP
PHP 搜索查询功能实现
2016/11/29 PHP
PHP实现的分页类定义与用法示例
2017/07/05 PHP
JQuery 将元素显示在屏幕的中央的代码
2010/02/27 Javascript
javascript 隐藏/显示指定的区域附HTML元素【legend】用法
2010/03/05 Javascript
js原型链原理看图说明
2012/07/07 Javascript
jQuery中的pushStack实现原理和应用实例
2015/02/03 Javascript
jQuery实现的背景动态变化导航菜单效果
2015/08/24 Javascript
基于javascript实现listbox左右移动
2016/01/29 Javascript
AngularJS中run方法的巧妙运用
2017/01/04 Javascript
微信小程序遇到修改数据后页面不渲染的问题解决
2017/03/09 Javascript
JavaScript获取tr td 的三种方式全面总结(推荐)
2017/08/15 Javascript
Vuejs 页面的区域化与组件封装的实现
2017/09/11 Javascript
Express下采用bcryptjs进行密码加密的方法
2018/02/07 Javascript
js实现聊天对话框
2020/02/08 Javascript
浅谈JavaScript窗体Window.ShowModalDialog使用
2020/07/22 Javascript
python访问类中docstring注释的实现方法
2015/05/04 Python
Python文件夹与文件的相关操作(推荐)
2016/07/25 Python
使用paramiko远程执行命令、下发文件的实例
2017/10/01 Python
AI人工智能 Python实现人机对话
2017/11/13 Python
python利用re,bs4,requests模块获取股票数据
2019/07/29 Python
Python散点图与折线图绘制过程解析
2019/11/30 Python
From CSV to SQLite3 by python 导入csv到sqlite实例
2020/02/14 Python
pandas使用之宽表变窄表的实现
2020/04/12 Python
python numpy实现rolling滚动案例
2020/06/08 Python
python绘制分布折线图的示例
2020/09/24 Python
如何设置PyCharm中的Python代码模版(推荐)
2020/11/20 Python
中东地区为妈妈们提供一切的头号购物目的地:Sprii
2018/05/06 全球购物
澳洲最大的时尚奢侈品电商平台:Cettire
2020/06/15 全球购物
医院护理人员的自我评价分享
2013/10/04 职场文书
windows11怎么查看自己安装的版本号? win11版本号的查看方法
2021/11/21 数码科技
SpringBoot整合minio快速入门教程(代码示例)
2022/04/03 Java/Android