ONNX LSTR Lane Detection
Python scripts for performing lane detection using the Lane Shape Prediction with Transformers (LSTR) model in ONNX.
- 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
Original Pytorch model
The pretrained Pytorch model was taken from the original repository.
- Image inference:
- Video inference:
Original video: https://youtu.be/2CIxM7x-Clc (by Yunfei Guo)