Onnx转ensorrt报错ERROR

Question

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

Answer 1

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

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

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