Convert Tensoflow V2 model ( pb format) to UFF

[AastaLLL]

Hi,

Sorry for the late update.

Could you try the suggestion in the below comment first:

tensorrt inference error while load onnx model

Hi, There may be a couple fixes for the output error. First can you check that the model was parsed correctly? parser->parse… should return True/False depending on whether it parsed the model successfully. Example: TensorRT/sampleINT8API.cpp at 07ed9b57b1ff7c24664388e5564b17f7ce2873e5 · NVIDIA/TensorRT · GitHub If the model was parsed successfully, then the output error has a pretty simple fix. Either mark the output with output_names=… in torch.onnx.export() when exporting your model to ONNX…

Thanks.