模型预测过程中报错 RuntimeError: Sizes of tensors must match except in dimension 1.

具体报错如上所示

问题背景

用gradio库写了一个YOLOv7检测logo的demo，上传图片之后报错如上。

解决方法

一开始我以为是类别数不对应的问题，然而并不是。

实际上是一个非常粗心的错误，原本的YOLOv7在训练和检测时都对图片进行了resize操作，而我在调用detect函数的时候，传入的图片是未经过resize的，所以就有了上述错误。

回YOLOv7源码可以看到它是

from utils.datasets import letterbox

然后调用letterbox函数实现resize，于是照搬，增加一行代码就好了。

当然自己写一段resize代码也是可以的。

模型预测过程中报错 RuntimeError: Sizes of tensors must match except in dimension 1.相关推荐

Pytorch_Geometric（PyG）使用DataLoader报错RuntimeError: Sizes of tensors must match except in dimension 0.
使用Pytorch_Geometric(PyG)时构建DataLoader,从DataLoader获取样本Batch时报错:RuntimeError: Sizes of tensors must ma ...
记录解决RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 27 but got size
问题描述在做目标检测服务过程中,将yolov7模型通过flask打包成预测服务API,此次训练的图像输入大小是1280,输入预测图片是如果图像大于1280则预测成功,小于1280则报RuntimeE ...
RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2
1. 背景介绍在训练语义分割模型时,出现以下错误: RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 a ...
RuntimeError: Sizes of tensors must match except in dimension 2. Got 5 and 4
调试代码的时候,遇到***RuntimeError: Sizes of tensors must match except in dimension 2. Got 5 and 4*** 这个问题, 解 ...
Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2
1. 背景介绍在训练编码解码网络模型时,出现以下错误: RuntimeError: Sizes of tensors must match except in dimension 1. ...
Sizes of tensors must match except in dimension 1. Expected size 24 but got size 25 for tensor numbe
在做图像分割的时候遇到了错误,错误如下: File "D:/segmentation/Pytorch-UNet-master/train.py", line 193, in < ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1.
最近在修改模型不支持的算子后,由于对模型的效果产生了影响,所以需要进行微调训练,但是训练的时候报错了,如下: File "/home/xxxxxxx/EdgeNets/nn_layers/e ...
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0—问题分析
声明: 这里仅仅提供一个思路,暂时没有绝对正确通用的解决方案作者:{ 墨理 } 感谢通过私信.评论交流,协力探讨解决问题的同学 UP .Penser 文章目录探讨的 pytorch 报错信息如下 ...
Pytorch踩坑: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0.
报错信息: RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 186 a ...

模型预测过程中报错 RuntimeError: Sizes of tensors must match except in dimension 1.

问题背景

解决方法

模型预测过程中报错 RuntimeError: Sizes of tensors must match except in dimension 1.相关推荐

最新文章

热门文章