yolov5推理的时候显示input size torch.size not equal to max model size

AssertionError: input size torch.sizecl1.3. 384, 640]) not equal to max model size (1, 3, 640, 640)
模型的尺寸不一样 要怎么解决

img

这个错误表明在进行推理时,输入的尺寸与模型期望的尺寸不匹配。解决此问题的步骤如下:

  1. 确保输入图像的尺寸与模型期望的尺寸匹配。根据你提供的错误信息来看,模型期望的输入尺寸是(1, 3, 640, 640)。如果你的输入图像尺寸不同于此, 将其调整为相应的大小。

  2. 如果你使用的是预训练的yolov5模型,并且没有更改过模型的网络结构或层设置,那么可以通过修改输入图像的尺寸来解决该问题。你可以通过调整图像的大小使其与模型期望的尺寸匹配。比如,你可以将输入图像的宽度和高度都调整为640像素。但要 注意:调整图像大小可能会导致一定的信息损失或形变,因此需要根据具体情况进行权衡。

  3. 如果你修改了yolov5模型的网络结构或层设置,确保所有的输入和输出尺寸匹配。在修改模型时,需特别注意输入层的尺寸是否与数据集和训练设置相符

你转到engine的时候是静态的,那么你的图片就要按静态的处理才行,也就是你需要指定imgs为[640,640]看看
另外,请检查一下这里的pt是否为false,正常情况下如果你是.engine的后缀的话,这里应该是False才对

img