Can we convert Detectron2 to TensorRT engine?

[amoolya.shetty]

I see that someone else has had the same problem -

Can't convert from torch-onnx to tensorrt

Hi, Sorry for the late update. We are trying to deploy the Detectron2 model with TensorRT. For the following progress, please find this topic: Thanks.

Detectron2 Jetson NX

Hi, This may work but not guarantee. Convert your detectron2 model into ONNX format first. You can find some information below: detectron2.export — detectron2 0.6 documentation Try to modify TensorRT FasterRCNN (caffe-based) example. Although the file format is different, you should get the required plugin layer from the example: https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleFasterRCNN Thanks.

And, you recommended modifying TensorRT FasterRCNN (caffe-based) example. (However, the GitHub link you provided is currently not working)

I also have a few other questions -

1. Are there any other ways to run the ONNX format on the TensorRT engine?
2. What other formats do the TensorRT engine support?