/tensorrt_smoke

3d object detection model smoke c++ inference code

Primary LanguageC++

tensorrt_smoke

3D检测模型SMOKE:https://github.com/lzccccc/SMOKE 的tensorrt推理代码.

tested on orin(jetpack5.0)

  • cuda11.4
  • tensorrt8.4.1.5
  • opencv4.5

usage

mkdir build
cd build 
cmake ../
make
./smoke_trt #第一次运行该程序时,将smoke_dla34.onnx转换为smoke_dla34.engine
./smoke_trt #使用smoke_dla34.engine做推理

performance

单纯模型推理时间:compute:54ms,fps:18 全流程时间:one frame:65ms,fps:15

Acknowledgement

TensorRT-SMOKE mmdetection3d