pytorch数据预处理错误的解决

更新时间：2020年02月20日 10:45:07 作者：慢行厚积

今天小编就为大家分享一篇pytorch数据预处理错误的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

出错：

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数据预处理错误的解决就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

pytest-fixture简介及其用法讲解
这篇文章主要介绍了pytest-fixture及其用法,最基本的用法就是一个fixture作为一个测试用例的参数传入，然后就可以在该测试用例中使用该fixture，需要的朋友可以参考下
2023-01-01
python3判断url链接是否为404的方法
这篇文章主要介绍了python3判断url链接是否为404的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-08-08
简单的python协同过滤程序实例代码
这篇文章主要介绍了简单的python协同过滤程序，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下
2018-01-01
python 画三维图像曲面图和散点图的示例
今天小编就为大家分享一篇python 画三维图像曲面图和散点图的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-12-12
Opencv对象追踪的示例代码
这篇文章主要介绍了Opencv对象追踪的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-03-03
用Python编写简单的gRPC服务的详细过程
gRPC 是可以在任何环境中运行的现代开源高性能 RPC 框架。接下来通过本文给大家介绍用Python编写简单的gRPC服务的详细过程，感兴趣的朋友一起看看吧
2021-07-07
Pandas中的loc与iloc区别与用法小结
loc函数：通过行索引 “Index” 中的具体值来取行数据（如取"Index"为"A"的行）而iloc函数：通过行号来取行数据（如取第二行的数据）,这篇文章介绍Pandas中的loc与iloc区别与用法,感兴趣的朋友一起看看吧
2024-01-01
解决PyCharm的Python.exe已经停止工作的问题
今天小编就为大家分享一篇解决PyCharm的Python.exe已经停止工作的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-11-11
pandas之query方法和sample随机抽样操作
这篇文章主要介绍了pandas之query方法和sample随机抽样操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-03-03
Python 探针的实现原理
本文将简单讲述一下 Python 探针的实现原理。同时为了验证这个原理，我们也会一起来实现一个简单的统计指定函数执行时间的探针程序。
2016-04-04