xuchangjia / tensorrt-laneatt

Code for TensorRT inference of LaneATT model

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

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