AssertionError: input size torch.Size([1, 3, 384, 640]) not equal to max model size (1, 3, 640, 640)

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就可以了，我这里调用的代码示例是下面这样：

然后成功推理。

(讨厌调维度😡