模型预测过程中报错 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代码也是可以的。