Onnx转ensorrt报错ERROR

Onnx-tensorrt报错ERROR: builtin_op_importers.cpp:3569 In function importUpsample:[8] Assertion failed: scales_input.is_weights()

这个错误通常是因为您在使用 onnx-tensorrt 时,Upsample 操作的输入 scales_input 不正确造成的。具体来说,这个 Assertion failed 的错误可能是因为 scales_input 不是 TensorRT 的 Weights 类型。

解决这个问题的方法通常有两种:

在导出 ONNX 模型时,确保将 Upsample 操作中的 scales 系数设置为常量,并将其设置为 Tensor 或 ISTensor。
在 TensorRT 中,使用合适的 API (如 setBindingDimensions 和 setDynamicDimensions) 显式地设置输入和输出的维度信息,以避免需要 TensorRT 推断维度大小的情况。
如果以上两种方法都无法解决这个问题,请考虑检查代码和环境配置是否正确,并尝试更新到最新版本的 ONNX 和 TensorRT 。