ONNX LSTR Lane Detection

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

Requirements

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

Installation

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.

Examples

• Image inference:

python image_lane_detection.py

• Video inference:

python video_lane_detection.py

Inference video Example

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