ONNX LSTR Lane Detection

Python scripts for performing lane detection using the Lane Shape Prediction with Transformers (LSTR) model in ONNX.

ONNX LSTR Lane Detection


  • Check the requirements.txt file. Additionally, pafy and youtube-dl are required for youtube video inference.


pip install -r requirements.txt
pip install pafy youtube-dl

ONNX model

The original model was converted to different formats (including .onnx) by PINTO0309, the models can be found in his repository.

Original Pytorch model

The pretrained Pytorch model was taken from the original repository.


  • Image inference:

python image_lane_detection.py
  • Video inference:

python video_lane_detection.py

Inference video Example

!ONNX LSTR Lane Detection on video

Original video: https://youtu.be/2CIxM7x-Clc (by Yunfei Guo)


View Github