Search code examples
tensorflowonnxnvidia-jetsontensorrt

Onnx to trt - [8] Assertion failed: creator && "Plugin not found

I am using TensorRT in order to convert a model from onnx to trt -format. The model is originally a model from Tensorflow Model Zoo (SSD ResNet50). When I try to convert it I get the error:

[E] [TRT] /home/jenkins/agent/workspace/OSS/OSS_L0_MergeRequest/oss/parsers/onnx/ModelImporter.cpp:708: ERROR: /home/jenkins/agent/workspace/OSS/OSS_L0_MergeRequest/oss/parsers/onnx/builtin_op_importers.cpp:4298 In function importFallbackPluginImporter: [8] Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?"
[E] Engine set up failed &&&& FAILED TensorRT.trtexec # trtexec --onnx=../model.onnx --fp16=enable --workspace=5500 --batch=1 --saveEngine=model_op11.trt --verbose

As far as I can tell it is looking for a plugin for the NonMaxSuppresion operation. Does anyone know how to convert a model from Tensorflow Model Zoo to TensorRT?

Solution

  • Got this fixed by using TensorRT 8.