ibaigorordo / onnx-lstr-lane-detection

Python scripts for performing lane detection using the LSTR model in ONNX

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)

onnx-lstr-lane-detection's Issues

postprocessing tuning

Hello,
I noticed the previews are only right in case of using LSTR_360X640 as the model type.
how could I edit it to work with other types?