Centernet mobilenet v2 from TFOD - failure when converting ONNX to TRT

[AastaLLL]

Hi,

This is a known issue due to the “IF” layer used in the model.
You can find some details below:

Resize error

Relevant Files Tensorflow model zoo Steps To Reproduce I downloaded the pre-trained model from Tensorflow model zoo. After that I tried running tf2onnx command: python3 -m tf2onnx.convert --saved-model models --output tf_model_op11.onnx --opset 11 This gave an onnx file with uint8 type. So I used Onnx graph surgeon tool to change the type: graph = gs.import_onnx(onnx.load("tf_model_op11.onnx")) for inp in graph.inputs: inp.dtype = np.float32 After getting the onnx file, I ran the comma…

To solve this, we provide an example to convert the EfficientDet with a customized parser.
It is one of the TensorFlow object detection models, and also gets stuck in the “IF” operation.

You can check the example below for the details:

Thanks.