ROS2 inference package based on mmdetection

I created a new sample package to make inference based on a trained model with mmdetection which is one of open source object detection toolbox based on PyTorch.
The training and inference pipelines are as below.

  1. train a model with mmdetection
  2. convert the model into end2end onnx file with mmdeploy
  3. do inference with the onnx file. It is parsed into execution file for TensorRT in my package.

Currently, yolox is only supported, but I would like to increase the supported models in the future.
Any feedback is welcome. Thank you.

1 Like