pytorch数据预处理错误的解决


Posted in Python onFebruary 20, 2020

出错:

Traceback (most recent call last):
 File "train.py", line 305, in <module>
 train_model(model_conv, criterion, optimizer_conv, exp_lr_scheduler)
 File "train.py", line 145, in train_model
 for inputs, age_labels, gender_labels in dataloaders[phase]:
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in __next__
 return self._process_next_batch(batch)
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch
 raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop
 samples = collate_fn([dataset[i] for i in batch_indices])
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in default_collate
 return [default_collate(samples) for samples in transposed]
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in <listcomp>
 return [default_collate(samples) for samples in transposed]
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 209, in default_collate
 return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 224 and 228 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307

这是因为输入的大小不匹配,跟数据集有关,也跟数据预处理中的函数相关:

transforms.Resize(input_size)

该函数是按比例缩放,可能是因为该数据集的分辨率不同,所以出来的结果不是(224,224)的,解决办法是改为使用:

transforms.Resize((input_size, input_size))

即可

以上这篇pytorch数据预处理错误的解决就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python HTMLParser模块解析html获取url实例
Apr 08 Python
将Django使用的数据库从MySQL迁移到PostgreSQL的教程
Apr 11 Python
Python中创建字典的几种方法总结(推荐)
Apr 27 Python
Python 中Pickle库的使用详解
Feb 24 Python
python得到电脑的开机时间方法
Oct 15 Python
python文件选择对话框的操作方法
Jun 27 Python
这可能是最好玩的python GUI入门实例(推荐)
Jul 19 Python
python编写一个会算账的脚本的示例代码
Jun 02 Python
python导入库的具体方法
Jun 18 Python
keras CNN卷积核可视化,热度图教程
Jun 22 Python
python爬取网页版QQ空间,生成各类图表
Jun 02 Python
如何利用python创作字符画
Jun 25 Python
Python异常继承关系和自定义异常实现代码实例
Feb 20 #Python
Python安装与卸载流程详细步骤(图解)
Feb 20 #Python
PyCharm 专业版安装图文教程
Feb 20 #Python
Python3.6安装卸载、执行命令、执行py文件的方法详解
Feb 20 #Python
Python 实现黑客帝国中的字符雨的示例代码
Feb 20 #Python
浅谈spring boot 集成 log4j 解决与logback冲突的问题
Feb 20 #Python
Python OpenCV读取显示视频的方法示例
Feb 20 #Python
You might like
php缓冲输出实例分析
2015/01/05 PHP
PHP单例模式实例分析【防继承,防克隆操作】
2019/05/22 PHP
Laravel5.4框架中视图共享数据的方法详解
2019/09/05 PHP
去除链接虚线全面分析总结
2006/08/15 Javascript
prototype 的说明 js类
2006/09/07 Javascript
slice函数的用法 之不错的应用
2006/12/29 Javascript
C#中TrimStart,TrimEnd,Trim在javascript上的实现
2011/01/17 Javascript
Jquery 复选框取值兼容FF和IE8(测试有效)
2013/10/29 Javascript
AngularJS+Node.js实现在线聊天室
2015/08/28 Javascript
JS实现带提示的星级评分效果完整实例
2015/10/30 Javascript
jQuery3.0中的buildFragment私有函数详解
2016/08/16 Javascript
jstree创建无限分级树的方法【基于ajax动态创建子节点】
2016/10/25 Javascript
js读取本地文件的实例
2017/12/22 Javascript
分析javascript原型及原型链
2018/03/18 Javascript
JS实现textarea通过换行或者回车把多行数字分割成数组并且去掉数组中空的值
2018/10/29 Javascript
微信小程序导航栏滑动定位功能示例(实现CSS3的positionsticky效果)
2019/01/24 Javascript
vue实现编辑器键盘抬起时内容跟随光标距顶位置向上滚动效果
2020/05/28 Javascript
Vue将props值实时传递 并可修改的操作
2020/08/09 Javascript
快速解决element的autofocus失效问题
2020/09/08 Javascript
利用js实现简易红绿灯
2020/10/15 Javascript
基于Cesium绘制抛物弧线
2020/11/18 Javascript
Windows中安装使用Virtualenv来创建独立Python环境
2016/05/31 Python
Python使用cookielib模块操作cookie的实例教程
2016/07/12 Python
简单谈谈python中的lambda表达式
2018/01/19 Python
python调用百度语音识别api
2018/08/30 Python
python高效过滤出文件夹下指定文件名结尾的文件实例
2018/10/21 Python
对python3 sort sorted 函数的应用详解
2019/06/27 Python
python 浅谈serial与stm32通信的编码问题
2019/12/18 Python
检测tensorflow是否使用gpu进行计算的方式
2020/02/03 Python
在Keras中利用np.random.shuffle()打乱数据集实例
2020/06/15 Python
法国滑雪假期的专家:Ski Planet
2019/11/02 全球购物
垃圾回收的优点和原理。并考虑2种回收机制
2016/10/16 面试题
武汉瑞得软件笔试题
2015/10/27 面试题
《奇妙的国际互联网》 教学反思
2014/02/25 职场文书
电子商务专业求职信范文
2015/03/19 职场文书
使用Redis实现点赞取消点赞的详细代码
2022/03/20 Redis