/TensorRT-LaneATT

Code for TensorRT inference of LaneATT model

Primary LanguagePythonMIT LicenseMIT

TensorRT-LaneATT

This repository creates a TensorRT engine of LaneATT for inference.

0.Prerequisites

  • TensorRT 8.0.3
  • CUDA 10.2
  • See install

1.ONNX

git clone https://github.com/lucastabelini/LaneATT.git
cd LaneATT
python laneatt_to_onnx.py

2.TensorRT

python onnx_to_tensorrt.py

sample

3.C++ Inference

cd laneatt
mkdir build && cd build
cmake ..
# modify image and engine path in laneatt_test.cc
make && ./laneatt_test

TODO

  • C++ inference
  • Measure speed on Xavier