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)
