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

作者：知新_RL | 2024-02-27 03:18:59

踩

runtimeerror: sizes of tensors must match except in dimension 1. expected si

具体报错如上所示

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

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

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

回YOLOv7源码可以看到它是

from utils.datasets import letterbox

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

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

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/149619