yolov5使用engine加速过程中报错:AssertionError: input size torch.Size([1, 3, 384, 640]) not equal to max model size (1, 3, 640, 640)
yolov5 6.1版本之后export文件自带导出onnx和engine(trt)代码,但是用生成的engine文件进行推理时总是报错:AssertionError: input size torch.Size([1, 3, 384, 640]) not equal to max model size (1, 3, 640, 640)
显示维度不同,在网上很少看到解决办法,试了一些已经存在的方法都不能解决我的问题,然后debug之后发现问题出在utils/augmentations.py文件中的letterbox函数中:
这个函数中将auto设置成了True,代表会强制转成640*640格式,所以我们只需要在调用这个函数的时候将auto参数设置为False,同时将scaleup参数设置为True就可以了,我这里调用的代码示例是下面这样:
然后成功推理。
(讨厌调维度😡