pytorch dataloader 取batch_size时候出现bug的解决方式
作者：U_C 发布时间：2023-08-12 01:27:45
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
来源：https://www.cnblogs.com/llfctt/p/11912884.html
0
投稿猜你喜欢
Mysql调优Explain工具详解以及实战演练 Explain工具介绍Explain分析示例explain 两个变种explain中的列- 通常python安装包都会被默认装在/usr/local/pythonx/lib/site-packages(linux)，但是我们有时想自
- Python 3 利用 Dlib 19.7 实现摄像头人脸检测特征点标定0.引言利用python开发，借助Dlib库捕获摄像头中的人脸，进行
- go fmt命令主要是用来帮你格式化所写好的代码文件【很多第三方集成软件都是使用了go fmt命令】一、使用：go fmt <文件名&
- 本文实例为大家分享了python爬取微信公众号文章的具体代码，供大家参考，具体内容如下# -*- coding: utf-8 -*-impo
- 前言此次的目标是爬取指定城市的天气预报信息，然后再用Python发送邮件到指定的邮箱。下面话不多说了，来一起看看详细的实现过程吧一、爬取天气
- 本文实例讲述了Python实现的简单hangman游戏。分享给大家供大家参考。具体如下：#!/usr/bin/env pythonimpor
- 先附上官方文档：https://pandas.pydata.org/pandas-docs/stable/reference/api/pan
- 客户的一台服务器出现Raid故障，经过我们的努力，数据基本恢复成功，但是其中的一个SQL Server日志文件（扩展名LDF）损坏严重，基本
- 如果是在Oracle10g之前，删除一个表空间中的数据文件后，其文件在数据库数据字典中会仍然存在，除非你删除表空间，否则文件信息不会清除。但
- 本文实例讲述了python简单文本处理的方法。分享给大家供大家参考。具体如下：由于有多线程的影响，c++项目打印出来的时间顺序不一致，导致不
- SQL Server 数据库定时自动备份，供大家参考，具体内容如下在SQL Server中出于数据安全的考虑，所以需要定期的备份数据库。而备
- KNN算法算是最简单的机器学习算法之一了，这个算法最大的特点是没有训练过程，是一种懒惰学习，这种结构也可以在tensorflow实现。KNN
- 我就废话不多说了，直接上代码吧！obj = Obj.objects.get(id=1)print obj.name #此时name的值假定为
- 注册模块default.asp 代码如下:<!DOCTYPE html PUBLIC "-//
- 有时，希望除去某些记录或更改它们的内容。DELETE 和 UPDATE 语句令我们能做到这一点。用update修改记录UPDATE tbl_
- Harris 角点检测算法1. 角点角点是水平方向、垂直方向变化都很大的像素。角点检测算法的基本思想：
- 零、本讲学习目标了解面向对象编程思想掌握类和对象的定义和使用了解Python中的对象一、面向对象（一）程序员“面向对象”在现实世界中存在各种
- 本文实例讲述了python实现的分析并统计nginx日志数据功能。分享给大家供大家参考，具体如下：利用python脚本分析nginx日志内容
- 前言在Django使用Celery异步发送邮件的过程中，遇到Celery日志提示任务已接收，但实际上任务并没有执行，解决后特此记录。使用版本
